diff options
-rw-r--r-- | ot/bregman.py | 2 |
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 |