diff options
author | Hicham Janati <hicham.janati@inria.fr> | 2019-09-03 17:35:23 +0200 |
---|---|---|
committer | Hicham Janati <hicham.janati@inria.fr> | 2019-09-03 17:35:23 +0200 |
commit | 49d9b5cf4eecefdc0fff4db6c43e85d16e478efb (patch) | |
tree | 2abaa14db67c81c7f7733b70fe8ac438b5072153 /ot/unbalanced.py | |
parent | 7efea812ad0b1c7e3783397dbd8f3ad802fb7ac2 (diff) |
fix doctest examples
Diffstat (limited to 'ot/unbalanced.py')
-rw-r--r-- | ot/unbalanced.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ot/unbalanced.py b/ot/unbalanced.py index 25e4cf5..d516dfc 100644 --- a/ot/unbalanced.py +++ b/ot/unbalanced.py @@ -326,7 +326,7 @@ def sinkhorn_knopp_unbalanced(a, b, M, reg, reg_m, numItermax=1000, >>> a=[.5, .5] >>> b=[.5, .5] >>> M=[[0., 1.],[1., 0.]] - >>> ot.unbalanced._sinkhorn_knopp_unbalanced(a, b, M, 1., 1.) + >>> ot.unbalanced.sinkhorn_knopp_unbalanced(a, b, M, 1., 1.) array([[0.51122823, 0.18807035], [0.18807035, 0.51122823]]) @@ -510,7 +510,7 @@ def sinkhorn_stabilized_unbalanced(a, b, M, reg, reg_m, tau=1e5, numItermax=1000 >>> a=[.5, .5] >>> b=[.5, .5] >>> M=[[0., 1.],[1., 0.]] - >>> ot.unbalanced._sinkhorn_stabilized_unbalanced(a, b, M, 1., 1.) + >>> ot.unbalanced.sinkhorn_stabilized_unbalanced(a, b, M, 1., 1.) array([[0.51122823, 0.18807035], [0.18807035, 0.51122823]]) |