diff options
author | Rémi Flamary <remi.flamary@gmail.com> | 2017-02-07 17:17:34 +0100 |
---|---|---|
committer | Rémi Flamary <remi.flamary@gmail.com> | 2017-02-07 17:17:34 +0100 |
commit | 0b806374d33ae83d39846096a1838b096c0c0b8e (patch) | |
tree | 66bc0255ec9bd25b59d1c941e54b299676929f42 /ot/bregman.py | |
parent | 4c344cfb57d6b32ae433777ae4d8f85c69d498e8 (diff) |
log return for sinkhorn
Diffstat (limited to 'ot/bregman.py')
-rw-r--r-- | ot/bregman.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/ot/bregman.py b/ot/bregman.py index fd22259..27f8ff5 100644 --- a/ot/bregman.py +++ b/ot/bregman.py @@ -139,6 +139,10 @@ def sinkhorn(a,b, M, reg, numItermax = 1000, stopThr=1e-9, verbose=False, log=Fa print('{:5s}|{:12s}'.format('It.','Err')+'\n'+'-'*19) print('{:5d}|{:8e}|'.format(cpt,err)) cpt = cpt +1 + if log: + log['u']=u + log['v']=v + #print 'err=',err,' cpt=',cpt if log: return u.reshape((-1,1))*K*v.reshape((1,-1)),log |