diff options
| author | Hicham Janati <hicham.janati@inria.fr> | 2019-07-23 10:32:41 +0200 |
|---|---|---|
| committer | Hicham Janati <hicham.janati@inria.fr> | 2019-07-23 10:32:41 +0200 |
| commit | 92292231a4d9661c399dbfd97b22d6f7f890f698 (patch) | |
| tree | 248945b95f655fd51997a895a61eeed24988aa72 /ot/unbalanced.py | |
| parent | 50a5a4111ada5e8c208da1acf731608930d0a278 (diff) | |
| parent | 0063cb87a10293a24ad1c9483be121745958c24a (diff) | |
Merge branch 'master' of https://github.com/rflamary/POT into fix_mismatch_error_94
Diffstat (limited to 'ot/unbalanced.py')
| -rw-r--r-- | ot/unbalanced.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/ot/unbalanced.py b/ot/unbalanced.py index 1453b31..14e9e36 100644 --- a/ot/unbalanced.py +++ b/ot/unbalanced.py @@ -395,7 +395,8 @@ def sinkhorn_knopp_unbalanced(a, b, M, reg, mu, numItermax=1000, print( '{:5s}|{:12s}'.format('It.', 'Err') + '\n' + '-' * 19) print('{:5d}|{:8e}|'.format(cpt, err)) - cpt = cpt + 1 + cpt += 1 + if log: log['logu'] = np.log(u + 1e-16) log['logv'] = np.log(v + 1e-16) |