diff options
Diffstat (limited to 'examples/plot_optim_OTreg.py')
-rw-r--r-- | examples/plot_optim_OTreg.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/examples/plot_optim_OTreg.py b/examples/plot_optim_OTreg.py index 0041770..1c3a1c5 100644 --- a/examples/plot_optim_OTreg.py +++ b/examples/plot_optim_OTreg.py @@ -31,7 +31,7 @@ M /= M.max() G0 = ot.emd(a, b, M) -pl.figure(3) +pl.figure(3, figsize=(5, 5)) ot.plot.plot1D_mat(a, b, G0, 'OT matrix G0') #%% Example with Frobenius norm regularization @@ -64,7 +64,7 @@ reg = 1e-3 Ge = ot.optim.cg(a, b, M, reg, f, df, verbose=True) -pl.figure(4) +pl.figure(4, figsize=(5, 5)) ot.plot.plot1D_mat(a, b, Ge, 'OT matrix Entrop. reg') #%% Example with Frobenius norm + entropic regularization with gcg @@ -82,6 +82,6 @@ reg2 = 1e-1 Gel2 = ot.optim.gcg(a, b, M, reg1, reg2, f, df, verbose=True) -pl.figure(5) +pl.figure(5, figsize=(5, 5)) ot.plot.plot1D_mat(a, b, Gel2, 'OT entropic + matrix Frob. reg') pl.show() |