diff options
Diffstat (limited to 'examples/others/plot_screenkhorn_1D.py')
-rw-r--r-- | examples/others/plot_screenkhorn_1D.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/examples/others/plot_screenkhorn_1D.py b/examples/others/plot_screenkhorn_1D.py index 2023649..3640b88 100644 --- a/examples/others/plot_screenkhorn_1D.py +++ b/examples/others/plot_screenkhorn_1D.py @@ -62,8 +62,8 @@ ot.plot.plot1D_mat(a, b, M, 'Cost matrix M') # Screenkhorn lambd = 2e-03 # entropy parameter -ns_budget = 30 # budget number of points to be keeped in the source distribution -nt_budget = 30 # budget number of points to be keeped in the target distribution +ns_budget = 30 # budget number of points to be kept in the source distribution +nt_budget = 30 # budget number of points to be kept in the target distribution G_screen = screenkhorn(a, b, M, lambd, ns_budget, nt_budget, uniform=False, restricted=True, verbose=True) pl.figure(4, figsize=(5, 5)) |