Skip to content

Commit

Permalink
delete plt sentences
Browse files Browse the repository at this point in the history
  • Loading branch information
zhyantao committed May 19, 2019
1 parent 36a9bf9 commit 7c54249
Showing 1 changed file with 0 additions and 17 deletions.
17 changes: 0 additions & 17 deletions train.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@
from keras.callbacks import EarlyStopping
import tensorflow as tf
import keras.backend.tensorflow_backend as K
import matplotlib.pyplot as plt

config = tf.ConfigProto()
config.gpu_options.allow_growth=True
Expand Down Expand Up @@ -48,19 +47,3 @@
validation_steps=20,
validation_data=valid_data,
callbacks=[model_checkpoint,tb_cb,early_stopping_monitor])

# draw the loss and accuracy curve
plt.figure(12, figsize=(6, 6), dpi=60)
plt.subplot(211)
plt.plot(history.history['loss'], label='train')
plt.plot(history.history['val_loss'], label='val')
plt.title('loss')
plt.legend()

plt.subplot(212)
plt.plot(history.history['acc'], label='train')
plt.plot(history.history['val_acc'], label='val')
plt.title('acc')
plt.legend()

plt.show()

0 comments on commit 7c54249

Please sign in to comment.