summaryrefslogtreecommitdiff
path: root/ot/lp/__init__.py
diff options
context:
space:
mode:
authorKilian Fatras <kilianfatras@dhcp-206-12-53-20.eduroam.wireless.ubc.ca>2018-08-28 17:24:07 -0700
committerKilian Fatras <kilianfatras@dhcp-206-12-53-20.eduroam.wireless.ubc.ca>2018-08-28 17:24:07 -0700
commite885d78cc9608d791a9d1561d2f4e0b783ba0761 (patch)
treee03a553873f110d1b8e0f15cc6f9248c916a405c /ot/lp/__init__.py
parent77b68901c5415ddc5d9ab5215a6fa97723de3de9 (diff)
debug sgd dual
Diffstat (limited to 'ot/lp/__init__.py')
-rw-r--r--ot/lp/__init__.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/ot/lp/__init__.py b/ot/lp/__init__.py
index 5dda82a..4c0d170 100644
--- a/ot/lp/__init__.py
+++ b/ot/lp/__init__.py
@@ -18,6 +18,8 @@ from .emd_wrap import emd_c, check_result
from ..utils import parmap
from .cvx import barycenter
+__all__=['emd', 'emd2', 'barycenter', 'cvx']
+
def emd(a, b, M, numItermax=100000, log=False):
"""Solves the Earth Movers distance problem and returns the OT matrix