aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorVasil Zlatanov <v@skozl.com>2019-03-10 18:46:24 +0000
committerVasil Zlatanov <v@skozl.com>2019-03-10 18:46:24 +0000
commit07ae3a62f460e8b1ae700194cd85f05b2d2f011d (patch)
tree46e38d904d58af68b4c75dfdec614733cf991090
parent717da7a7e173cd971baddd9f8edb6de668b1e815 (diff)
downloade4-gan-07ae3a62f460e8b1ae700194cd85f05b2d2f011d.tar.gz
e4-gan-07ae3a62f460e8b1ae700194cd85f05b2d2f011d.tar.bz2
e4-gan-07ae3a62f460e8b1ae700194cd85f05b2d2f011d.zip
Add padding to plot_probas
-rw-r--r--lenet.py1
1 files changed, 1 insertions, 0 deletions
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()