aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authornunzip <np.scarh@gmail.com>2019-02-27 23:16:08 +0000
committernunzip <np.scarh@gmail.com>2019-02-27 23:16:08 +0000
commitd0f97c01830b1018b1327461c1503bc1cf316eae (patch)
treee81ebee869255582aba6bf7d2845369c0aab6937
parent723ac9bb7094b482a2efc6bc97139409d2bde9c4 (diff)
downloade4-gan-d0f97c01830b1018b1327461c1503bc1cf316eae.tar.gz
e4-gan-d0f97c01830b1018b1327461c1503bc1cf316eae.tar.bz2
e4-gan-d0f97c01830b1018b1327461c1503bc1cf316eae.zip
Plot accuracy instead of loss
-rw-r--r--lenet.py2
1 files changed, 1 insertions, 1 deletions
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):