summaryrefslogtreecommitdiff
path: root/ot/gromov.py
diff options
context:
space:
mode:
authorncourty <ncourty@irisa.fr>2017-09-13 10:07:47 +0200
committerncourty <ncourty@irisa.fr>2017-09-13 10:07:47 +0200
commit5a2ebfab92a655e636efee1d91d44c3023e25828 (patch)
treef28941495cb3b8f80ce0f17f789e81be7b3fb552 /ot/gromov.py
parent84c272394d41d159d07174306b324590b3ffe40c (diff)
Corrections on Gromov
Diffstat (limited to 'ot/gromov.py')
-rw-r--r--ot/gromov.py4
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