summaryrefslogtreecommitdiff
path: root/ot/lp/cvx.py
diff options
context:
space:
mode:
authorRémi Flamary <remi.flamary@gmail.com>2018-05-29 16:08:33 +0200
committerRémi Flamary <remi.flamary@gmail.com>2018-05-29 16:08:33 +0200
commit54f0b47e55c966d5492e4ce19ec4e704ef3278d6 (patch)
treef04e912395fda68a0ee39d7db1442d0dd3efba46 /ot/lp/cvx.py
parent3f1482238925932bb6c9c606651427491a65365c (diff)
update documentation for barycenter function
Diffstat (limited to 'ot/lp/cvx.py')
-rw-r--r--ot/lp/cvx.py4
1 files changed, 2 insertions, 2 deletions
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