diff options
author | Rémi Flamary <remi.flamary@gmail.com> | 2018-05-31 13:45:11 +0200 |
---|---|---|
committer | Rémi Flamary <remi.flamary@gmail.com> | 2018-05-31 13:45:11 +0200 |
commit | 8046b8c424d7b80f520e212e2bf8de41cb624aab (patch) | |
tree | d679ae2ffe48501b5c95284261364d0ec11b07dd | |
parent | d370f18aaac4ed6903e3b4251c8b9c4685c53cc3 (diff) |
pep8
-rw-r--r-- | README.md | 2 | ||||
-rw-r--r-- | ot/smooth.py | 2 |
2 files changed, 2 insertions, 2 deletions
@@ -16,7 +16,7 @@ It provides the following solvers: * OT Network Flow solver for the linear program/ Earth Movers Distance [1]. * Entropic regularization OT solver with Sinkhorn Knopp Algorithm [2] and stabilized version [9][10] with optional GPU implementation (requires cudamat). * Smooth optimal transport solvers (dual and semi-dual) for KL and squared L2 regularization [17]. -* Non regularized Wasserstein barycenters [16] with LP solver. +* Non regularized Wasserstein barycenters [16] with LP solver (only small scale). * Bregman projections for Wasserstein barycenter [3] and unmixing [4]. * Optimal transport for domain adaptation with group lasso regularization [5] * Conditional gradient [6] and Generalized conditional gradient for regularized OT [7]. diff --git a/ot/smooth.py b/ot/smooth.py index dd734b3..87d61ef 100644 --- a/ot/smooth.py +++ b/ot/smooth.py @@ -580,7 +580,7 @@ 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, + alpha, res = solve_semi_dual(a, b, M, regul, max_iter=numItermax, tol=stopThr, verbose=verbose) # reconstruct transport matrix |