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')
~