persim.sliced_wasserstein

persim.sliced_wasserstein(PD1, PD2, M=50)[source]

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