diff options
| author | ievred <ievgen.redko@univ-st-etienne.fr> | 2020-04-01 09:13:58 +0200 |
|---|---|---|
| committer | ievred <ievgen.redko@univ-st-etienne.fr> | 2020-04-01 09:13:58 +0200 |
| commit | 547a03ef87e4aa92edc1e89ee2db04114e1a8ad5 (patch) | |
| tree | f6795752c32fd95879324fc59ab280d3cb0b2551 /ot/bregman.py | |
| parent | 439860609df786a877383775dd901afe28480cc9 (diff) | |
fix test example add M to log
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) |