From 48a6fa973b65a696d087af35c1b410d73ecddde4 Mon Sep 17 00:00:00 2001 From: nunzip Date: Wed, 27 Feb 2019 23:49:55 +0000 Subject: Fix accuracy plot --- lenet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lenet.py b/lenet.py index fa27422..2372857 100644 --- a/lenet.py +++ b/lenet.py @@ -116,7 +116,7 @@ def train_classifier(x_train, y_train, x_val, y_val, batch_size=128, epochs=100, model.compile(loss='categorical_crossentropy', metrics=metrics, optimizer=optimizer) history = model.fit(x_train, y_train, batch_size=batch_size, epochs=epochs, verbose=1, validation_data = (x_val, y_val)) - plot_history(history, metric=metrics) + plot_history(history, metrics) return model def test_classifier(model, x_test, y_true): -- cgit v1.2.3