https://medium.com/@monocosmo77/how-to-work-with-wasserstein-distance-part4-machine-learning-816017c1498