From: Francois Fleuret <francois@fleuret.org>
Date: Tue, 30 Jun 2020 09:13:57 +0000 (+0200)
Subject: Added the legend in dd-multi-mse.pdf
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=c75c1288649c1d4bbd941ba040367deffb5feccb;p=pytorch.git

Added the legend in dd-multi-mse.pdf
---

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)