summaryrefslogtreecommitdiff
path: root/src/python/example/diagram_vectorizations_distances_kernels.py
diff options
context:
space:
mode:
authormathieu <mathieu.carriere3@gmail.com>2020-02-13 16:11:34 -0500
committermathieu <mathieu.carriere3@gmail.com>2020-02-13 16:11:34 -0500
commitd9290a78741fc14dc0f87d395da967a4d561b34a (patch)
tree3266f9921ff690502d27481bfffb3625245c8430 /src/python/example/diagram_vectorizations_distances_kernels.py
parent7327abc115ae3e06a512782ec5833783086c3866 (diff)
small modif on example file
Diffstat (limited to 'src/python/example/diagram_vectorizations_distances_kernels.py')
-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)