diff options
Diffstat (limited to 'src/python/example/diagram_vectorizations_distances_kernels.py')
| -rwxr-xr-x | src/python/example/diagram_vectorizations_distances_kernels.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/src/python/example/diagram_vectorizations_distances_kernels.py b/src/python/example/diagram_vectorizations_distances_kernels.py index 119072eb..de22d9e7 100755 --- a/src/python/example/diagram_vectorizations_distances_kernels.py +++ b/src/python/example/diagram_vectorizations_distances_kernels.py @@ -117,6 +117,16 @@ X = SW.fit(diags) Y = SW.transform(diags2) print("SW kernel is " + str(Y[0][0])) +W = WassersteinDistance(order=2, internal_p=2, mode="pot") +X = W.fit(diags) +Y = W.transform(diags2) +print("Wasserstein distance (POT) is " + str(Y[0][0])) + +W = WassersteinDistance(order=2, internal_p=2, mode="hera", delta=0.0001) +X = W.fit(diags) +Y = W.transform(diags2) +print("Wasserstein distance (hera) is " + str(Y[0][0])) + W = BottleneckDistance(epsilon=.001) X = W.fit(diags) Y = W.transform(diags2) |