From c75c1288649c1d4bbd941ba040367deffb5feccb Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Tue, 30 Jun 2020 11:13:57 +0200 Subject: [PATCH] Added the legend in dd-multi-mse.pdf --- ddpol.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/ddpol.py b/ddpol.py index 6ace38d..1c5d131 100755 --- a/ddpol.py +++ b/ddpol.py @@ -132,6 +132,8 @@ for nb_train_samples in range(nb_train_samples_min, nb_train_samples_max + 1, 2) ax.plot(torch.arange(args.D_max + 1), mse_train, color = (e, e, 1.0), label = f'Train N={nb_train_samples}') ax.plot(torch.arange(args.D_max + 1), mse_test, color = (1.0, e, e), label = f'Test N={nb_train_samples}') +ax.legend(frameon = False) + fig.savefig('dd-multi-mse.pdf', bbox_inches='tight') plt.close(fig) -- 2.20.1