https://medium.com/@monocosmo77/how-to-work-with-wasserstein-distance-part2-machine-learning-14bff6180d11