From fae3543a1a5ee731dd4f961a860c856521f7f3fd Mon Sep 17 00:00:00 2001 From: Vasil Zlatanov Date: Wed, 27 Feb 2019 19:02:35 +0000 Subject: Fix categorical_accuracy --- lenet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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) -- cgit v1.2.3