diff options
author | ngayraud <nat.gayraud@gmail.com> | 2019-08-12 15:49:25 -0400 |
---|---|---|
committer | ngayraud <nat.gayraud@gmail.com> | 2019-08-12 15:49:25 -0400 |
commit | 092866815cf906012f9194b87af1e7ae0270f7e7 (patch) | |
tree | b98dfb4a0ac9165d22f8276f6f3a481e8b316f97 /ot/unbalanced.py | |
parent | b2157e9b3458388571f6ae87d80f47f500dfa166 (diff) |
Added Unbalaced transport to domain adaptation methods. Corrected small bug related to warnings in unbalaced.py . Added an error message when user wants to normalize with other than expected cost normalization functions.
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 |