diff options
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 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) |