Skip to content

Instantly share code, notes, and snippets.

@prl900

prl900/plot_tf_history.py

Last active October 19, 2018 04:11
Show Gist options
  • Save prl900/302d9219406427de4c1797143d4c8960 to your computer and use it in GitHub Desktop.
Save prl900/302d9219406427de4c1797143d4c8960 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
plot_tf_history.py
import pickle
import matplotlib
matplotlib.use('agg')
import matplotlib.pyplot as plt
#hist = pickle.load(open("trainHistoryDict_0-5-7", 'rb'))
hist = pickle.load(open("trainHistoryDict_9", 'rb'))
print hist.keys()
plt.plot(hist['loss'])
plt.plot(hist['val_loss'])
plt.title('model loss')
plt.ylabel('loss')
plt.xlabel('epoch')
plt.legend(['train', 'test'], loc='upper left')
plt.savefig('foo9.png')
~
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment