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
|
# -*- coding: utf-8 -*-
"""
Solvers related to Gromov-Wasserstein problems.
"""
# Author: Remi Flamary <remi.flamary@unice.fr>
# Cedric Vincent-Cuaz <cedvincentcuaz@gmail.com>
#
# License: MIT License
# All submodules and packages
from ._utils import (init_matrix, tensor_product, gwloss, gwggrad,
update_square_loss, update_kl_loss, update_feature_matrix,
init_matrix_semirelaxed)
from ._gw import (gromov_wasserstein, gromov_wasserstein2,
fused_gromov_wasserstein, fused_gromov_wasserstein2,
solve_gromov_linesearch, gromov_barycenters, fgw_barycenters)
from ._bregman import (entropic_gromov_wasserstein,
entropic_gromov_wasserstein2,
entropic_gromov_barycenters,
entropic_fused_gromov_wasserstein,
entropic_fused_gromov_wasserstein2,
entropic_fused_gromov_barycenters)
from ._estimators import (GW_distance_estimation, pointwise_gromov_wasserstein,
sampled_gromov_wasserstein)
from ._semirelaxed import (semirelaxed_gromov_wasserstein,
semirelaxed_gromov_wasserstein2,
semirelaxed_fused_gromov_wasserstein,
semirelaxed_fused_gromov_wasserstein2,
solve_semirelaxed_gromov_linesearch,
entropic_semirelaxed_gromov_wasserstein,
entropic_semirelaxed_gromov_wasserstein2,
entropic_semirelaxed_fused_gromov_wasserstein,
entropic_semirelaxed_fused_gromov_wasserstein2)
from ._dictionary import (gromov_wasserstein_dictionary_learning,
gromov_wasserstein_linear_unmixing,
fused_gromov_wasserstein_dictionary_learning,
fused_gromov_wasserstein_linear_unmixing)
__all__ = ['init_matrix', 'tensor_product', 'gwloss', 'gwggrad', 'update_square_loss',
'update_kl_loss', 'update_feature_matrix', 'init_matrix_semirelaxed',
'gromov_wasserstein', 'gromov_wasserstein2', 'fused_gromov_wasserstein',
'fused_gromov_wasserstein2', 'solve_gromov_linesearch', 'gromov_barycenters',
'fgw_barycenters', 'entropic_gromov_wasserstein', 'entropic_gromov_wasserstein2',
'entropic_gromov_barycenters', 'entropic_fused_gromov_wasserstein',
'entropic_fused_gromov_wasserstein2', 'entropic_fused_gromov_barycenters',
'GW_distance_estimation', 'pointwise_gromov_wasserstein', 'sampled_gromov_wasserstein',
'semirelaxed_gromov_wasserstein', 'semirelaxed_gromov_wasserstein2',
'semirelaxed_fused_gromov_wasserstein', 'semirelaxed_fused_gromov_wasserstein2',
'solve_semirelaxed_gromov_linesearch', 'entropic_semirelaxed_gromov_wasserstein',
'entropic_semirelaxed_gromov_wasserstein2', 'entropic_semirelaxed_fused_gromov_wasserstein',
'entropic_semirelaxed_fused_gromov_wasserstein2', 'gromov_wasserstein_dictionary_learning',
'gromov_wasserstein_linear_unmixing', 'fused_gromov_wasserstein_dictionary_learning',
'fused_gromov_wasserstein_linear_unmixing']
|
