summaryrefslogtreecommitdiff
path: root/ot/__init__.py
blob: 0602eedff530510344dca78222f3fff56d65f9b7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
# Python Optimal Transport toolbox

# All submodules and packages
from . import lp 
from . import bregman
from . import optim 
from . import utils
from . import datasets
from . import plot
from . import da



# OT functions
from .lp import emd
from .bregman import sinkhorn,barycenter
from .da import sinkhorn_lpl1_mm

# utils functions
from .utils import dist,unif

__all__ = ["emd","sinkhorn","utils",'datasets','bregman','lp','plot','dist','unif','barycenter','sinkhorn_lpl1_mm','da','optim']