diff options
author | Nicolas Courty <Nico@MacBook-Pro-de-Nicolas.local> | 2017-09-13 01:09:26 +0200 |
---|---|---|
committer | Nicolas Courty <Nico@MacBook-Pro-de-Nicolas.local> | 2017-09-13 01:09:26 +0200 |
commit | 55db3508917c73c4811a82933f892fc3017300f2 (patch) | |
tree | f28941495cb3b8f80ce0f17f789e81be7b3fb552 | |
parent | 84c272394d41d159d07174306b324590b3ffe40c (diff) |
Corrections on Gromov
-rw-r--r-- | ot/gromov.py | 4 |
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
|