diff options
Diffstat (limited to 'ot/bregman.py')
-rw-r--r-- | ot/bregman.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ot/bregman.py b/ot/bregman.py index d17aaf0..fb959e9 100644 --- a/ot/bregman.py +++ b/ot/bregman.py @@ -1603,6 +1603,7 @@ def jcpot_barycenter(Xs, Ys, Xt, reg, metric='sqeuclidean', numItermax=100, # build the cost matrix and the Gibbs kernel M = dist(Xs[d], Xt, metric=metric) M = M / np.median(M) + dom['M'] = M K = np.empty(M.shape, dtype=M.dtype) np.divide(M, -reg, out=K) |