scipy.stats.wasserstein_distance
函数只返回两个输入分布p
和q
之间的最小距离(解(。但该距离是距离矩阵和最优传输矩阵乘积的结果,该矩阵必须在同一函数内计算。
如何提取与解决方案相对应的距离矩阵和最优传输矩阵作为第二和第三输出自变量?
您似乎无法从scipy的wasserstein_dance中获得计算出的传输矩阵。不过,你可以通过其他包获得,比如https://github.com/wmayner/pyemd.我已经使用这个包一段时间了,它运行得很好,同时执行速度也很快。查看Usage部分中的函数emd_with_flow((。
则距离矩阵是EMD计算的输入,而不是输出。