From 07ae3a62f460e8b1ae700194cd85f05b2d2f011d Mon Sep 17 00:00:00 2001
From: Vasil Zlatanov <v@skozl.com>
Date: Sun, 10 Mar 2019 18:46:24 +0000
Subject: Add padding to plot_probas

---
 lenet.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/lenet.py b/lenet.py
index 881cfd7..0fe8277 100644
--- a/lenet.py
+++ b/lenet.py
@@ -144,6 +144,7 @@ def train_classifier(x_train, y_train, x_val, y_val, batch_size=128, epochs=100,
   return model 
 
 def plot_probas(model, x_test, y_true):
+    x_test = np.pad(x_test, ((0,0),(2,2),(2,2),(0,0)), 'constant')
     probas = model.predict(x_test)
     skplt.metrics.plot_roc(y_true, probas)
     plt.show()
-- 
cgit v1.2.3-54-g00ecf