From ffdce041b6570e6a57c356e7028da4f199649a96 Mon Sep 17 00:00:00 2001 From: lan-qing Date: Fri, 21 Feb 2020 21:32:29 +0800 Subject: [PATCH 1/2] A small typo It seems the labels of "test" and "train" are wrong. --- train_BayesByBackprop_MNIST.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_BayesByBackprop_MNIST.py b/train_BayesByBackprop_MNIST.py index 874e674..5302b47 100644 --- a/train_BayesByBackprop_MNIST.py +++ b/train_BayesByBackprop_MNIST.py @@ -216,7 +216,7 @@ plt.xlabel('epoch') plt.grid(b=True, which='major', color='k', linestyle='-') plt.grid(b=True, which='minor', color='k', linestyle='--') -lgd = plt.legend(['test error', 'train error'], markerscale=marker, prop={'size': textsize, 'weight': 'normal'}) +lgd = plt.legend(['train error', 'test error'], markerscale=marker, prop={'size': textsize, 'weight': 'normal'}) ax = plt.gca() plt.title('classification costs') for item in ([ax.title, ax.xaxis.label, ax.yaxis.label] + From 59ba197bf34765862ac6648987e7520fd1b59a3e Mon Sep 17 00:00:00 2001 From: lan-qing Date: Fri, 21 Feb 2020 21:33:39 +0800 Subject: [PATCH 2/2] a small typo --- train_BootrapEnsemble_MNIST.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train_BootrapEnsemble_MNIST.py b/train_BootrapEnsemble_MNIST.py index e85fb0b..bde6681 100644 --- a/train_BootrapEnsemble_MNIST.py +++ b/train_BootrapEnsemble_MNIST.py @@ -207,7 +207,7 @@ plt.xlabel('epoch') plt.grid(b=True, which='major', color='k', linestyle='-') plt.grid(b=True, which='minor', color='k', linestyle='--') - lgd = plt.legend(['test error', 'train error'], markerscale=marker, prop={'size': textsize, 'weight': 'normal'}) + lgd = plt.legend(['train error', 'test error'], markerscale=marker, prop={'size': textsize, 'weight': 'normal'}) ax = plt.gca() plt.title('classification costs') for item in ([ax.title, ax.xaxis.label, ax.yaxis.label] + @@ -235,4 +235,4 @@ plt.savefig(results_dir + '/err%d.png' % iii, bbox_extra_artists=(lgd,), box_inches='tight') -save_object(weight_set_samples, models_dir+'/state_dicts.pkl') \ No newline at end of file +save_object(weight_set_samples, models_dir+'/state_dicts.pkl')