diff options
| author | tlacombe <lacombe1993@gmail.com> | 2020-06-11 12:15:22 +0200 |
|---|---|---|
| committer | tlacombe <lacombe1993@gmail.com> | 2020-06-11 12:15:22 +0200 |
| commit | 4fe1d853d32846b8f34699c7121b4d7edfeb48ac (patch) | |
| tree | e102aba089131a58b81ba186696e152110034fdf /src/python/gudhi/wasserstein/wasserstein.py | |
| parent | 48d9ff9986cb3a6e36f60142784c2a02317dd327 (diff) | |
added file
Diffstat (limited to 'src/python/gudhi/wasserstein/wasserstein.py')
| -rw-r--r-- | src/python/gudhi/wasserstein/wasserstein.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/python/gudhi/wasserstein/wasserstein.py b/src/python/gudhi/wasserstein/wasserstein.py index 89ecab1c..6d7ec65f 100644 --- a/src/python/gudhi/wasserstein/wasserstein.py +++ b/src/python/gudhi/wasserstein/wasserstein.py @@ -88,7 +88,7 @@ def _perstot(X, order, internal_p, enable_autodiff): return np.linalg.norm(_dist_to_diag(X, internal_p), ord=order) -def wasserstein_distance(X, Y, matching=False, order=2., internal_p=2., enable_autodiff=False): +def wasserstein_distance(X, Y, matching=False, order=1., internal_p=np.inf, enable_autodiff=False): ''' :param X: (n x 2) numpy.array encoding the (finite points of the) first diagram. Must not contain essential points (i.e. with infinite coordinate). |