diff options
| author | Gard Spreemann <gspr@nonempty.org> | 2020-07-09 08:50:11 +0200 |
|---|---|---|
| committer | Gard Spreemann <gspr@nonempty.org> | 2020-07-09 08:50:11 +0200 |
| commit | 616c6039614e96b6cfd88eb7db25ce11c7302c30 (patch) | |
| tree | a745ec24604f660b256dbaee214affa497c9c21e /test/test_unbalanced.py | |
| parent | d62f05daf348b4e554056f298c66cbd64f5e3c6e (diff) | |
| parent | a16b9471d7114ec08977479b7249efe747702b97 (diff) | |
Merge branch 'dfsg/latest' into debian/sid
Diffstat (limited to 'test/test_unbalanced.py')
| -rw-r--r-- | test/test_unbalanced.py | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/test/test_unbalanced.py b/test/test_unbalanced.py index ca1efba..dfeaad9 100644 --- a/test/test_unbalanced.py +++ b/test/test_unbalanced.py @@ -31,9 +31,11 @@ def test_unbalanced_convergence(method): G, log = ot.unbalanced.sinkhorn_unbalanced(a, b, M, reg=epsilon, reg_m=reg_m, method=method, - log=True) + log=True, + verbose=True) loss = ot.unbalanced.sinkhorn_unbalanced2(a, b, M, epsilon, reg_m, - method=method) + method=method, + verbose=True) # check fixed point equations # in log-domain fi = reg_m / (reg_m + epsilon) @@ -73,7 +75,8 @@ def test_unbalanced_multiple_inputs(method): loss, log = ot.unbalanced.sinkhorn_unbalanced(a, b, M, reg=epsilon, reg_m=reg_m, method=method, - log=True) + log=True, + verbose=True) # check fixed point equations # in log-domain fi = reg_m / (reg_m + epsilon) |