diff --git a/demo.py b/demo.py index 0bccea43a1828d755c1210e8ef4bbba168c026e4..67d826c35d78ae10ce89d3c4f81e36db3ff7624c 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'])