persim.sliced_wasserstein

persim.sliced_wasserstein(PD1, PD2, M=50)

Implementation of Sliced Wasserstein distance as described in Sliced Wasserstein Kernel for Persistence Diagrams by Mathieu Carriere, Marco Cuturi, Steve Oudot (https://arxiv.org/abs/1706.03358)

Parameters:

• PD1 (np.array size (m,2)) – Persistence diagram

• PD2 (np.array size (n,2)) – Persistence diagram

• M (int, default is 50) – Iterations to run approximation.

Returns:

sw (float) – Sliced Wasserstein distance between PD1 and PD2