diff options
Diffstat (limited to 'ot/__init__.py')
-rw-r--r-- | ot/__init__.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/ot/__init__.py b/ot/__init__.py index 4c3c40e..87119e5 100644 --- a/ot/__init__.py +++ b/ot/__init__.py @@ -5,15 +5,17 @@ from . import utils from . import datasets from . import plot from . import bregman +from . import lp from . import da from . import optim + # OT functions -from ot.emd import emd +from ot.lp import emd from ot.bregman import sinkhorn,barycenter from ot.da import sinkhorn_lpl1_mm # utils functions from utils import dist,unif -__all__ = ["emd","sinkhorn","utils",'datasets','bregman','plot','dist','unif','barycenter','sinkhorn_lpl1_mm','da','optim'] +__all__ = ["emd","sinkhorn","utils",'datasets','bregman','lp','plot','dist','unif','barycenter','sinkhorn_lpl1_mm','da','optim'] |