From d0f97c01830b1018b1327461c1503bc1cf316eae Mon Sep 17 00:00:00 2001
From: nunzip <np.scarh@gmail.com>
Date: Wed, 27 Feb 2019 23:16:08 +0000
Subject: Plot accuracy instead of loss

---
 lenet.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lenet.py b/lenet.py
index 57ce218..495deaf 100644
--- a/lenet.py
+++ b/lenet.py
@@ -115,7 +115,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)
+  plot_history(history, metric=metrics)
   return model 
 
 def test_classifier(model, x_test, y_true):
-- 
cgit v1.2.3