diff options
author | Hicham Janati <hicham.janati@inria.fr> | 2019-06-18 22:26:48 +0200 |
---|---|---|
committer | Hicham Janati <hicham.janati@inria.fr> | 2019-06-18 22:26:48 +0200 |
commit | adf9d046445bf142a29d914352f397b36f7905c0 (patch) | |
tree | e13c7231390288fafd6eb09702db1cde6521a81f /examples/plot_UOT_1D.py | |
parent | 897982718a5fd81a9a591d80a7d50839399fc088 (diff) |
update Readme + minor rendering in examples
Diffstat (limited to 'examples/plot_UOT_1D.py')
-rw-r--r-- | examples/plot_UOT_1D.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/examples/plot_UOT_1D.py b/examples/plot_UOT_1D.py index 59b7e77..2ea8b05 100644 --- a/examples/plot_UOT_1D.py +++ b/examples/plot_UOT_1D.py @@ -1,8 +1,8 @@ # -*- coding: utf-8 -*- """ -==================== +=============================== 1D Unbalanced optimal transport -==================== +=============================== This example illustrates the computation of Unbalanced Optimal transport using a Kullback-Leibler relaxation. @@ -53,7 +53,7 @@ pl.plot(x, a, 'b', label='Source distribution') pl.plot(x, b, 'r', label='Target distribution') pl.legend() -#%% plot distributions and loss matrix +# plot distributions and loss matrix pl.figure(2, figsize=(5, 5)) ot.plot.plot1D_mat(a, b, M, 'Cost matrix M') @@ -64,7 +64,7 @@ ot.plot.plot1D_mat(a, b, M, 'Cost matrix M') # -------------- -#%% Sinkhorn +# Sinkhorn epsilon = 0.1 # entropy parameter alpha = 1. # Unbalanced KL relaxation parameter |