1 files changed, 1 insertions, 0 deletions

diff --git a/lenet.py b/lenet.py
index 982cd7e..4752fc3 100644
--- a/lenet.py
+++ b/lenet.py
@@ -20,6 +20,7 @@ def import_mnist():
   X_test, y_test = mnist.test.images, mnist.test.labels
   y_train = keras.utils.to_categorical(y_train, 10)
   y_validation = keras.utils.to_categorical(y_validation, 10)  
+  y_test = keras.utils.to_categorical(y_test, 10)  
   return X_train, y_train, X_validation, y_validation, X_test, y_test
 
 def plot_images(images, cls_true, cls_pred=None):