From 2c9f992157844d6253a302905417e86580ac6b12 Mon Sep 17 00:00:00 2001 From: ievred Date: Tue, 7 Apr 2020 13:50:11 +0200 Subject: upd --- ot/bregman.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'ot/bregman.py') diff --git a/ot/bregman.py b/ot/bregman.py index ec81924..61dfa52 100644 --- a/ot/bregman.py +++ b/ot/bregman.py @@ -1608,7 +1608,7 @@ def jcpot_barycenter(Xs, Ys, Xt, reg, metric='sqeuclidean', numItermax=100, # build the cost matrix and the Gibbs kernel Mtmp = dist(Xs[d], Xt, metric=metric) Mtmp = Mtmp / np.median(Mtmp) - M.append(M) + M.append(Mtmp) Ktmp = np.empty(Mtmp.shape, dtype=Mtmp.dtype) np.divide(Mtmp, -reg, out=Ktmp) -- cgit v1.2.3