summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorVincent Rouvreau <10407034+VincentRouvreau@users.noreply.github.com>2020-05-11 08:42:40 +0200
committerGitHub <noreply@github.com>2020-05-11 08:42:40 +0200
commit627772e4c5bc7038b0814182dbb918b08356c892 (patch)
tree024c05fd1d7199b0c76af4c6f68f025b762854cf
parent484732c8ad30721ba4fa596bcb8a3835ad3bc431 (diff)
Fixed by @tlacombe
Co-authored-by: Théo Lacombe <lacombe1993@gmail.com>
-rw-r--r--src/python/gudhi/wasserstein/barycenter.py5
1 files changed, 1 insertions, 4 deletions
diff --git a/src/python/gudhi/wasserstein/barycenter.py b/src/python/gudhi/wasserstein/barycenter.py
index 1cf8edb3..7eeeae7a 100644
--- a/src/python/gudhi/wasserstein/barycenter.py
+++ b/src/python/gudhi/wasserstein/barycenter.py
@@ -52,9 +52,7 @@ def lagrangian_barycenter(pdiagset, init=None, verbose=False):
Namely, ``G[k] = [...(i, j)...]``, where ``(i,j)`` indicates that ``pdiagset[k][i]`` is matched to ``Y[j]``
if ``i = -1`` or ``j = -1``, it means they represent the diagonal.
- - `"energy"`, ``float`` representing the Frechet energy value obtained.
-
- It is the mean of squared distances of observations to the output.
+ - `"energy"`, ``float`` representing the Frechet energy value obtained. It is the mean of squared distances of observations to the output.
- `"nb_iter"`, ``int`` number of iterations performed before convergence of the algorithm.
'''
@@ -149,4 +147,3 @@ def lagrangian_barycenter(pdiagset, init=None, verbose=False):
return Y, log
else:
return Y
-