diff options
author | mathieu <mathieu.carriere3@gmail.com> | 2020-01-16 17:02:55 -0500 |
---|---|---|
committer | mathieu <mathieu.carriere3@gmail.com> | 2020-01-16 17:02:55 -0500 |
commit | 85ceea9512634a62664208cd2d0f1ce48bafa171 (patch) | |
tree | 16ece19cf25d11fd875b2599f72c09007e469b98 /src/python/example | |
parent | cabc43b34723efa7640313348b844eabe9971e38 (diff) |
added wasserstein class
Diffstat (limited to 'src/python/example')
-rwxr-xr-x | src/python/example/diagram_vectorizations_distances_kernels.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/src/python/example/diagram_vectorizations_distances_kernels.py b/src/python/example/diagram_vectorizations_distances_kernels.py index 119072eb..66c32cc2 100755 --- a/src/python/example/diagram_vectorizations_distances_kernels.py +++ b/src/python/example/diagram_vectorizations_distances_kernels.py @@ -9,7 +9,7 @@ from gudhi.representations import DiagramSelector, Clamping, Landscape, Silhouet TopologicalVector, DiagramScaler, BirthPersistenceTransform,\ PersistenceImage, PersistenceWeightedGaussianKernel, Entropy, \ PersistenceScaleSpaceKernel, SlicedWassersteinDistance,\ - SlicedWassersteinKernel, BottleneckDistance, PersistenceFisherKernel + SlicedWassersteinKernel, BottleneckDistance, WassersteinDistance, PersistenceFisherKernel D = np.array([[0.,4.],[1.,2.],[3.,8.],[6.,8.], [0., np.inf], [5., np.inf]]) diags = [D] @@ -117,6 +117,11 @@ X = SW.fit(diags) Y = SW.transform(diags2) print("SW kernel is " + str(Y[0][0])) +W = WassersteinDistance(order=2, internal_p=2) +X = W.fit(diags) +Y = W.transform(diags2) +print("Wasserstein distance is " + str(Y[0][0])) + W = BottleneckDistance(epsilon=.001) X = W.fit(diags) Y = W.transform(diags2) |