summaryrefslogtreecommitdiff
path: root/ot
diff options
context:
space:
mode:
authorRémi Flamary <remi.flamary@gmail.com>2018-05-11 16:07:03 +0200
committerRémi Flamary <remi.flamary@gmail.com>2018-05-11 16:07:03 +0200
commit94eb4a24fe09c8a193933d8c7f48a657da4e6c52 (patch)
treefb0f29304f844f472646a99cf6051bee9b8744ff /ot
parent27032b6fa0f2f68af3fe4f90e5dcbb68f130a962 (diff)
update documentation in bregman
Diffstat (limited to 'ot')
-rw-r--r--ot/bregman.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/ot/bregman.py b/ot/bregman.py
index 07b8660..9c84aed 100644
--- a/ot/bregman.py
+++ b/ot/bregman.py
@@ -844,6 +844,8 @@ def barycenter(A, M, reg, weights=None, numItermax=1000,
loss matrix for OT
reg : float
Regularization term >0
+ weights : np.ndarray (n,)
+ Weights of each histogram i_i on the simplex
numItermax : int, optional
Max number of iterations
stopThr : float, optional