summaryrefslogtreecommitdiff
path: root/examples/plot_optim_OTreg.py
diff options
context:
space:
mode:
authorAlexandre Gramfort <alexandre.gramfort@m4x.org>2017-07-12 22:20:08 +0200
committerAlexandre Gramfort <alexandre.gramfort@m4x.org>2017-07-20 14:05:12 +0200
commitd6091dae858a82f69e6843859f164269fa338c6b (patch)
tree4c6b60ef30514a7d86d06b0f7664769c7d871bb6 /examples/plot_optim_OTreg.py
parentc5a72cc1ff6a1f5462fdca6ced837180820fe19b (diff)
more
Diffstat (limited to 'examples/plot_optim_OTreg.py')
-rw-r--r--examples/plot_optim_OTreg.py6
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()