summaryrefslogtreecommitdiff
path: root/src/python/example
diff options
context:
space:
mode:
Diffstat (limited to 'src/python/example')
-rwxr-xr-xsrc/python/example/diagram_vectorizations_distances_kernels.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/src/python/example/diagram_vectorizations_distances_kernels.py b/src/python/example/diagram_vectorizations_distances_kernels.py
index 6352d2b5..507ead7c 100755
--- a/src/python/example/diagram_vectorizations_distances_kernels.py
+++ b/src/python/example/diagram_vectorizations_distances_kernels.py
@@ -120,12 +120,12 @@ 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 is " + str(Y[0][0]))
+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 is " + str(Y[0][0]))
+print("Wasserstein distance (hera) is " + str(Y[0][0]))
W = BottleneckDistance(epsilon=.001)
X = W.fit(diags)