diff options
| author | mathieu <mathieu.carriere3@gmail.com> | 2020-02-13 16:01:29 -0500 |
|---|---|---|
| committer | mathieu <mathieu.carriere3@gmail.com> | 2020-02-13 16:01:29 -0500 |
| commit | ef0f82ef2155440827e17c552abb49b509866fc7 (patch) | |
| tree | c012fa25aed54c23bb3f4c6cc5c271af7d1982c9 /src/python/example/diagram_vectorizations_distances_kernels.py | |
| parent | 2f2db197a38e45ac4fe01dec0c029171c251029b (diff) | |
integrated hera
Diffstat (limited to 'src/python/example/diagram_vectorizations_distances_kernels.py')
| -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])) |