From 7816075cd507d14ef09df13afbc4be525bedd08c Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Wed, 18 Dec 2019 17:24:04 +0100 Subject: [PATCH] OCD update. --- denoising-ae-field.py | 17 +++++++++++------ 1 file changed, 11 insertions(+), 6 deletions(-) diff --git a/denoising-ae-field.py b/denoising-ae-field.py index 67d2415..cc4af81 100755 --- a/denoising-ae-field.py +++ b/denoising-ae-field.py @@ -13,7 +13,7 @@ model = nn.Sequential( nn.Linear(100, 2) ) -############################################################ +###################################################################### def data_zigzag(nb): a = torch.empty(nb).uniform_(0, 1).view(-1, 1) @@ -80,12 +80,17 @@ ax.set_xlim(-1.6, 1.6) ax.set_ylim(-1.6, 1.6) ax.set_aspect(1) -plot_field = ax.quiver(origins[:, 0].numpy(), origins[:, 1].numpy(), - field[:, 0].numpy(), field[:, 1].numpy(), - units = 'xy', scale = 1, - width = 3e-3, headwidth = 25, headlength = 25) +plot_field = ax.quiver( + origins[:, 0].numpy(), origins[:, 1].numpy(), + field[:, 0].numpy(), field[:, 1].numpy(), + units = 'xy', scale = 1, + width = 3e-3, headwidth = 25, headlength = 25 +) -plot_data = ax.scatter(data[:, 0].numpy(), data[:, 1].numpy(), s = 1, color = 'tab:blue') +plot_data = ax.scatter( + data[:, 0].numpy(), data[:, 1].numpy(), + s = 1, color = 'tab:blue' +) fig.savefig('denoising_field.pdf', bbox_inches='tight') -- 2.39.5