summaryrefslogtreecommitdiff
path: root/ot/smooth.py
diff options
context:
space:
mode:
authorRémi Flamary <remi.flamary@gmail.com>2018-05-31 13:03:40 +0200
committerRémi Flamary <remi.flamary@gmail.com>2018-05-31 13:03:40 +0200
commited0d4171c6291a15360bdb8a955b0783585da749 (patch)
treebc9e63e628ce3138d3ad8547115559ab7b1496a2 /ot/smooth.py
parent10f9b0d1b02c2b5f4c4eeac0c1f803657c89764b (diff)
update readme
Diffstat (limited to 'ot/smooth.py')
-rw-r--r--ot/smooth.py9
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)