aboutsummaryrefslogtreecommitdiff
path: root/lenet.py
diff options
context:
space:
mode:
Diffstat (limited to 'lenet.py')
-rw-r--r--lenet.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/lenet.py b/lenet.py
index a62138b..982cd7e 100644
--- a/lenet.py
+++ b/lenet.py
@@ -101,7 +101,7 @@ def plot_history(history, metric = None):
plt.ylabel('Loss')
plt.xlabel('Epoch')
-def train_classifier(x_train, y_train, x_val, y_val, batch_size=128, epochs=100, metrics=['categorical_accuracy'], optimizer = None):
+def train_classifier(x_train, y_train, x_val, y_val, batch_size=128, epochs=100, metrics=[categorical_accuracy], optimizer = None):
shape = (32, 32, 1)
# Pad data to 32x32 (MNIST is 28x28)