diff options
Diffstat (limited to 'examples/plot_optim_OTreg.py')
-rw-r--r-- | examples/plot_optim_OTreg.py | 5 |
1 files changed, 3 insertions, 2 deletions
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) |