diff options
author | RĂ©mi Flamary <remi.flamary@gmail.com> | 2019-08-21 14:43:58 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2019-08-21 14:43:58 +0200 |
commit | abfe183a49caaf74a07e595ac40920dae05a3c22 (patch) | |
tree | 4d5fe2d98d249a252cda18db29ff47edc472a2ab /ot/unbalanced.py | |
parent | b2157e9b3458388571f6ae87d80f47f500dfa166 (diff) | |
parent | ce86d1476b32771d32b7e55566e7cab45bb57b3a (diff) |
Merge pull request #100 from ngayraud/add_unbalanced_da
[MRG] Adds Unbalaced transport to domain adaptation methods + bugfixes
Diffstat (limited to 'ot/unbalanced.py')
-rw-r--r-- | ot/unbalanced.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/unbalanced.py b/ot/unbalanced.py index 467fda2..0f0692e 100644 --- a/ot/unbalanced.py +++ b/ot/unbalanced.py @@ -364,7 +364,7 @@ def sinkhorn_knopp_unbalanced(a, b, M, reg, alpha, numItermax=1000, or np.any(np.isinf(u)) or np.any(np.isinf(v))): # we have reached the machine precision # come back to previous solution and quit loop - warnings.warn('Numerical errors at iteration', cpt) + warnings.warn('Numerical errors at iteration %s' % cpt) u = uprev v = vprev break |