1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
# -*- coding: utf-8 -*-
"""
Solvers related to Bregman projections for entropic regularized OT
"""
# Author: Remi Flamary <remi.flamary@unice.fr>
# Cédric Vincent-Cuaz <cedvincentcuaz@gmail.com>
#
# License: MIT License
from ._utils import geometricBar, geometricMean, projR, projC
from ._sinkhorn import (
sinkhorn,
sinkhorn2,
sinkhorn_knopp,
sinkhorn_log,
greenkhorn,
sinkhorn_stabilized,
sinkhorn_epsilon_scaling,
)
from ._barycenter import (
barycenter,
barycenter_sinkhorn,
free_support_sinkhorn_barycenter,
barycenter_stabilized,
barycenter_debiased,
jcpot_barycenter,
)
from ._convolutional import (
convolutional_barycenter2d,
convolutional_barycenter2d_debiased,
)
from ._empirical import (
empirical_sinkhorn,
empirical_sinkhorn2,
empirical_sinkhorn_divergence,
)
from ._screenkhorn import screenkhorn
from ._dictionary import unmix
from ._geomloss import empirical_sinkhorn2_geomloss, geomloss
__all__ = [
"geometricBar",
"geometricMean",
"projR",
"projC",
"sinkhorn",
"sinkhorn2",
"sinkhorn_knopp",
"sinkhorn_log",
"greenkhorn",
"sinkhorn_stabilized",
"sinkhorn_epsilon_scaling",
"barycenter",
"barycenter_sinkhorn",
"free_support_sinkhorn_barycenter",
"barycenter_stabilized",
"barycenter_debiased",
"jcpot_barycenter",
"convolutional_barycenter2d",
"convolutional_barycenter2d_debiased",
"empirical_sinkhorn",
"empirical_sinkhorn2",
"empirical_sinkhorn2_geomloss",
"empirical_sinkhorn_divergence",
"geomloss",
"screenkhorn",
"unmix",
]
|
