summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-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
-