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