From ec67362de5ec785e3871eac75a8aa477857092c4 Mon Sep 17 00:00:00 2001 From: RĂ©mi Flamary Date: Wed, 30 Aug 2017 20:57:52 +0200 Subject: pep8 --- examples/plot_OT_1D.py | 2 +- examples/plot_optim_OTreg.py | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/examples/plot_OT_1D.py b/examples/plot_OT_1D.py index be6f5b3..77114c4 100644 --- a/examples/plot_OT_1D.py +++ b/examples/plot_OT_1D.py @@ -52,7 +52,7 @@ pl.figure(2, figsize=(5, 5)) ot.plot.plot1D_mat(a, b, M, 'Cost matrix M') ############################################################################## -# Solve EMD +# Solve EMD ############################################################################## #%% EMD diff --git a/examples/plot_optim_OTreg.py b/examples/plot_optim_OTreg.py index d36b269..7ef6a6b 100644 --- a/examples/plot_optim_OTreg.py +++ b/examples/plot_optim_OTreg.py @@ -13,7 +13,7 @@ import ot ############################################################################## -# Generate data +# Generate data ############################################################################## #%% parameters @@ -32,7 +32,7 @@ M = ot.dist(x.reshape((n, 1)), x.reshape((n, 1))) M /= M.max() ############################################################################## -# Solve EMD +# Solve EMD ############################################################################## #%% EMD @@ -92,6 +92,7 @@ ot.plot.plot1D_mat(a, b, Ge, 'OT matrix Entrop. reg') #%% Example with Frobenius norm + entropic regularization with gcg + def f(G): return 0.5 * np.sum(G**2) -- cgit v1.2.3