diff options
author | RĂ©mi Flamary <remi.flamary@gmail.com> | 2020-06-22 11:59:10 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-06-22 11:59:10 +0200 |
commit | acfff525e043c8733936e8e99367543edd28822b (patch) | |
tree | 9cc7eb227e2fc42363d9b95aab1ed448a39d158f /ot/bregman.py | |
parent | b2eae39e8a422b18ecc3fadc08bc909ee1dae55f (diff) |
Revert "[WIP] Update bregman.py : barycenter_sinkhorn function (#195)" (#196)
This reverts commit b2eae39e8a422b18ecc3fadc08bc909ee1dae55f.
Diffstat (limited to 'ot/bregman.py')
-rw-r--r-- | ot/bregman.py | 2 |
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 |