summaryrefslogtreecommitdiff
path: root/ot/bregman.py
diff options
context:
space:
mode:
Diffstat (limited to 'ot/bregman.py')
-rw-r--r--ot/bregman.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/bregman.py b/ot/bregman.py
index 457bdd4..f1f8437 100644
--- a/ot/bregman.py
+++ b/ot/bregman.py
@@ -1116,7 +1116,7 @@ def barycenter_sinkhorn(A, M, reg, weights=None, numItermax=1000,
err = 1
UKv = np.dot(K, np.divide(A.T, np.sum(K, axis=0)).T)
- u = (geometricBar(weights, UKv) / UKv.T).T
+ u = (geometricMean(UKv) / UKv.T).T
while (err > stopThr and cpt < numItermax):
cpt = cpt + 1