https://medium.com/@monocosmo77/how-to-work-with-wasserstein-distance-part3-machine-learning-90a5eb85f8ff