diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2016-10-24 16:34:26 +0200 |
|---|---|---|
| committer | Rémi Flamary <remi.flamary@gmail.com> | 2016-10-24 16:34:26 +0200 |
| commit | 2f2bee41dcc2685a2ee9adfd60253ebb3131e5e8 (patch) | |
| tree | 95f484fb63ab90211e52efe4098f0c9b595390d5 /ot/bregman.py | |
| parent | e009eac25a50e1fb26294a739f9c778e3a693c85 (diff) | |
update demo barycenter
Diffstat (limited to 'ot/bregman.py')
| -rw-r--r-- | ot/bregman.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ot/bregman.py b/ot/bregman.py index e46506b..8362dc4 100644 --- a/ot/bregman.py +++ b/ot/bregman.py @@ -136,7 +136,7 @@ def barycenter(A,M,reg, weights=None, numItermax = 1000, tol_error=1e-4,log=dict log['all_err'].append(err) log['niter']=cpt - return geometricBar(weights,UKv) + return geometricBar(weights,UKv),log def unmixBregman(distrib,D,M,M0,h0,reg,reg0,alpha,numItermax = 1000, tol_error=1e-3,log=dict()): @@ -181,4 +181,4 @@ def unmixBregman(distrib,D,M,M0,h0,reg,reg0,alpha,numItermax = 1000, tol_error=1 log['niter']=cpt - return np.sum(K0,axis=1) + return np.sum(K0,axis=1),log |