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

# All submodules and packages
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.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','lp','plot','dist','unif','barycenter','sinkhorn_lpl1_mm','da','optim']