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