diff options
author | Hicham Janati <hicham.janati@inria.fr> | 2019-07-22 15:28:59 +0200 |
---|---|---|
committer | Hicham Janati <hicham.janati@inria.fr> | 2019-07-22 15:28:59 +0200 |
commit | 50a5a4111ada5e8c208da1acf731608930d0a278 (patch) | |
tree | 7ecffa81938598fcf8f19345ce48919f4c53eab8 /ot/unbalanced.py | |
parent | 5c0ed104b2890c609bdadfe0fcb0e836ba7a6ef1 (diff) |
fix doctest examples
Diffstat (limited to 'ot/unbalanced.py')
-rw-r--r-- | ot/unbalanced.py | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/ot/unbalanced.py b/ot/unbalanced.py index ca24e8b..1453b31 100644 --- a/ot/unbalanced.py +++ b/ot/unbalanced.py @@ -77,8 +77,8 @@ def sinkhorn_unbalanced(a, b, M, reg, mu, method='sinkhorn', numItermax=1000, >>> b=[.5, .5] >>> M=[[0., 1.], [1., 0.]] >>> ot.sinkhorn_unbalanced(a, b, M, 1, 1) - array([[0.51122823, 0.18807035], - [0.18807035, 0.51122823]]) + array([[0.51122818, 0.18807034], + [0.18807034, 0.51122818]]) References @@ -193,7 +193,7 @@ def sinkhorn_unbalanced2(a, b, M, reg, mu, method='sinkhorn', >>> b=[.5, .5] >>> M=[[0., 1.],[1., 0.]] >>> ot.unbalanced.sinkhorn_unbalanced2(a, b, M, 1., 1.) - array([0.31912866]) + array([0.31912862]) @@ -308,8 +308,8 @@ def sinkhorn_knopp_unbalanced(a, b, M, reg, mu, numItermax=1000, >>> b=[.5, .5] >>> M=[[0., 1.],[1., 0.]] >>> ot.unbalanced.sinkhorn_knopp_unbalanced(a, b, M, 1., 1.) - array([[0.51122823, 0.18807035], - [0.18807035, 0.51122823]]) + array([[0.51122818, 0.18807034], + [0.18807034, 0.51122818]]) References ---------- @@ -479,8 +479,8 @@ def sinkhorn_stabilized_unbalanced(a, b, M, reg, mu, tau=1e5, numItermax=1000, >>> b=[.5, .5] >>> M=[[0., 1.],[1., 0.]] >>> ot.unbalanced.sinkhorn_stabilized_unbalanced(a, b, M, 1., 1.) - array([[0.51122823, 0.18807035], - [0.18807035, 0.51122823]]) + array([[0.51122818, 0.18807034], + [0.18807034, 0.51122818]]) References ---------- |