diff options
| author | ievred <ievgen.redko@univ-st-etienne.fr> | 2020-04-07 13:50:11 +0200 |
|---|---|---|
| committer | ievred <ievgen.redko@univ-st-etienne.fr> | 2020-04-07 13:50:11 +0200 |
| commit | 2c9f992157844d6253a302905417e86580ac6b12 (patch) | |
| tree | d7314b752b8d50f7bcc8e34eab11e2a109ddb222 /ot | |
| parent | 34e13d467e376e9bfee2eb15771d9308518c2adb (diff) | |
upd
Diffstat (limited to 'ot')
| -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) |