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

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

# OT functions
from emd 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','plot','dist','unif','barycenter','sinkhorn_lpl1_mm','da','optim']