From 1343beb4c94b932c2876a5a0323a789851564a92 Mon Sep 17 00:00:00 2001 From: forbesjon2 <forbesjon2@huskers.unl.edu> Date: Sat, 12 Oct 2019 00:29:55 +0000 Subject: [PATCH] Update demo.py --- demo.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/demo.py b/demo.py index 0bccea4..67d826c 100644 --- a/demo.py +++ b/demo.py @@ -64,8 +64,7 @@ with tf.Session() as session: # train & validate. Allocate 20% for validation set. 100 epochs - history = model.fit(np.array(data_train), np.array(labels_train), batch_size = 32, epochs=100, callbacks=[early_stopping_ -callback], verbose=1, shuffle=True, validation_split=0.2) + history = model.fit(np.array(data_train), np.array(labels_train), batch_size = 32, epochs=100, callbacks=[early_stopping_callback], verbose=1, shuffle=True, validation_split=0.2) # plot plt.plot(history.history['loss']) plt.plot(history.history['val_loss']) -- GitLab