diff options
author | Rémi Flamary <remi.flamary@gmail.com> | 2018-05-31 13:03:40 +0200 |
---|---|---|
committer | Rémi Flamary <remi.flamary@gmail.com> | 2018-05-31 13:03:40 +0200 |
commit | ed0d4171c6291a15360bdb8a955b0783585da749 (patch) | |
tree | bc9e63e628ce3138d3ad8547115559ab7b1496a2 /ot/smooth.py | |
parent | 10f9b0d1b02c2b5f4c4eeac0c1f803657c89764b (diff) |
update readme
Diffstat (limited to 'ot/smooth.py')
-rw-r--r-- | ot/smooth.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/ot/smooth.py b/ot/smooth.py index 1a9972b..b3649e9 100644 --- a/ot/smooth.py +++ b/ot/smooth.py @@ -22,6 +22,7 @@ #OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF #THE POSSIBILITY OF SUCH DAMAGE. + # Author: Mathieu Blondel # Remi Flamary <remi.flamary@unice.fr> @@ -31,6 +32,13 @@ Smooth and Sparse Optimal Transport. Mathieu Blondel, Vivien Seguy, Antoine Rolet. In Proc. of AISTATS 2018. https://arxiv.org/abs/1710.06276 + +[17] Blondel, M., Seguy, V., & Rolet, A. (2018). Smooth and Sparse Optimal +Transport. Proceedings of the Twenty-First International Conference on +Artificial Intelligence and Statistics (AISTATS). + +Original code from https://github.com/mblondel/smooth-ot/ + """ import numpy as np @@ -402,6 +410,7 @@ def get_plan_from_semi_dual(alpha, b, C, regul): def smooth_ot_dual(a, b, M, reg, reg_type='l2', method="L-BFGS-B", stopThr=1e-9, numItermax=500, log=False): + if reg_type.lower() in ['l2', 'squaredl2']: regul = SquaredL2(gamma=reg) |