diff options
author | Hicham Janati <hicham.janati@inria.fr> | 2019-06-12 17:06:32 +0200 |
---|---|---|
committer | Hicham Janati <hicham.janati@inria.fr> | 2019-06-12 17:06:32 +0200 |
commit | 11381a7ecc79ef719ee9107167c3adc22b5a3f59 (patch) | |
tree | fd2b3b7c4ae59bc4050e6b69579f940e2a5a5f18 /examples/plot_UOT_1D.py | |
parent | 28b549ef3ef93c01462cd811d6e55c36ae5a76a2 (diff) |
integrate comments of jmassich
Diffstat (limited to 'examples/plot_UOT_1D.py')
-rw-r--r-- | examples/plot_UOT_1D.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/examples/plot_UOT_1D.py b/examples/plot_UOT_1D.py index 1b1dd9c..59b7e77 100644 --- a/examples/plot_UOT_1D.py +++ b/examples/plot_UOT_1D.py @@ -66,9 +66,9 @@ ot.plot.plot1D_mat(a, b, M, 'Cost matrix M') #%% Sinkhorn -lambd = 0.1 -alpha = 1. -Gs = ot.unbalanced.sinkhorn_unbalanced(a, b, M, lambd, alpha, verbose=True) +epsilon = 0.1 # entropy parameter +alpha = 1. # Unbalanced KL relaxation parameter +Gs = ot.unbalanced.sinkhorn_unbalanced(a, b, M, epsilon, alpha, verbose=True) pl.figure(4, figsize=(5, 5)) ot.plot.plot1D_mat(a, b, Gs, 'UOT matrix Sinkhorn') |