From 55db3508917c73c4811a82933f892fc3017300f2 Mon Sep 17 00:00:00 2001 From: Nicolas Courty Date: Wed, 13 Sep 2017 01:09:26 +0200 Subject: Corrections on Gromov --- ot/gromov.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'ot') 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 -- cgit v1.2.3