Corrections on Gromov

1 files changed, 2 insertions, 2 deletions

diff --git a/ot/gromov.py b/ot/gromov.py
index 7968e5e..20bf7ee 100644
--- a/ot/gromov.py
+++ b/ot/gromov.py
@@ -158,7 +158,7 @@ def update_square_loss(p, lambdas, T, Cs):
     Parameters

     ----------

     p  : ndarray, shape (N,)

-         weights in the targeted barycenter

+         masses in the targeted barycenter

     lambdas : list of float

               list of the S spaces' weights

     T : list of S np.ndarray(ns,N)

@@ -401,7 +401,7 @@ def gromov_barycenters(N, Cs, ps, p, lambdas, loss_fun, epsilon,
          weights in the targeted barycenter

     lambdas : list of float

               list of the S spaces' weights

-    L :  tensor-matrix multiplication function based on specific loss function

+    loss_fun :  tensor-matrix multiplication function based on specific loss function

     update : function(p,lambdas,T,Cs) that updates C according to a specific Kernel

              with the S Ts couplings calculated at each iteration

     epsilon : float