summaryrefslogtreecommitdiff
path: root/ot/__init__.py
blob: 4f4489e4990ff54b4c693fc3b196cdb76635e3b6 (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']