aboutsummaryrefslogtreecommitdiff
path: root/lenet.py
diff options
context:
space:
mode:
Diffstat (limited to 'lenet.py')
-rw-r--r--lenet.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/lenet.py b/lenet.py
index 4752fc3..c9ef78c 100644
--- a/lenet.py
+++ b/lenet.py
@@ -20,7 +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)
+ 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):
@@ -128,6 +128,6 @@ def test_classifier(model, x_test, y_true):
# If file run directly, perform quick test
if __name__ == '__main__':
- x_train, y_train, x_val, y_val, _, _ = import_mnist()
+ x_train, y_train, x_val, y_val, x_t, y_t = import_mnist()
model = train_classifier(x_train[:100], y_train[:100], x_val, y_val, epochs=1)
- test_classifier(model, x_val, y_val)
+ test_classifier(model, x_t, y_t)