diff options
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 66c32cc2..6352d2b5 100755 --- a/src/python/example/diagram_vectorizations_distances_kernels.py +++ b/src/python/example/diagram_vectorizations_distances_kernels.py @@ -117,7 +117,12 @@ X = SW.fit(diags) Y = SW.transform(diags2) print("SW kernel is " + str(Y[0][0])) -W = WassersteinDistance(order=2, internal_p=2) +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])) + +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])) |