diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2018-05-31 13:40:45 +0200 |
|---|---|---|
| committer | Rémi Flamary <remi.flamary@gmail.com> | 2018-05-31 13:40:45 +0200 |
| commit | d370f18aaac4ed6903e3b4251c8b9c4685c53cc3 (patch) | |
| tree | 2716bab79217cb6cff3b0fae56aef6b1a69dc6a2 /ot/smooth.py | |
| parent | fb883fcddff31cc4e21c921ebeb9a550eaa48ff0 (diff) | |
bug verbose semi-dual
Diffstat (limited to 'ot/smooth.py')
| -rw-r--r-- | ot/smooth.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/ot/smooth.py b/ot/smooth.py index 4c46e73..dd734b3 100644 --- a/ot/smooth.py +++ b/ot/smooth.py @@ -580,10 +580,11 @@ def smooth_ot_semi_dual(a, b, M, reg, reg_type='l2', method="L-BFGS-B", stopThr= raise NotImplementedError('Unknown regularization') # solve dual - alpha, res = solve_semi_dual(a, b, M, regul, max_iter=numItermax, tol=stopThr) + alpha, res = solve_semi_dual(a, b, M, regul, max_iter=numItermax, + tol=stopThr, verbose=verbose) # reconstruct transport matrix - G = get_plan_from_semi_dual(alpha, b, M, regul, verbose=verbose) + G = get_plan_from_semi_dual(alpha, b, M, regul) if log: log = {'alpha': alpha, 'res': res} |