diff options
-rw-r--r-- | ot/bregman.py | 4 | ||||
-rw-r--r-- | ot/lp/cvx.py | 4 |
2 files changed, 4 insertions, 4 deletions
diff --git a/ot/bregman.py b/ot/bregman.py index e788ef5..b017c1a 100644 --- a/ot/bregman.py +++ b/ot/bregman.py @@ -839,13 +839,13 @@ def barycenter(A, M, reg, weights=None, numItermax=1000, Parameters ---------- A : np.ndarray (d,n) - n training distributions of size d + n training distributions a_i of size d M : np.ndarray (d,d) loss matrix for OT reg : float Regularization term >0 weights : np.ndarray (n,) - Weights of each histogram i_i on the simplex + Weights of each histogram a_i on the simplex (barycentric coodinates) numItermax : int, optional Max number of iterations stopThr : float, optional diff --git a/ot/lp/cvx.py b/ot/lp/cvx.py index fe9ac76..c8c75bc 100644 --- a/ot/lp/cvx.py +++ b/ot/lp/cvx.py @@ -46,13 +46,13 @@ def barycenter(A, M, weights=None, verbose=False, log=False, solver='interior-po Parameters ---------- A : np.ndarray (d,n) - n training distributions of size d + n training distributions a_i of size d M : np.ndarray (d,d) loss matrix for OT reg : float Regularization term >0 weights : np.ndarray (n,) - Weights of each histogram i_i on the simplex + Weights of each histogram a_i on the simplex (barycentric coodinates) verbose : bool, optional Print information along iterations log : bool, optional |