Kerasのmodel.fitが返すhistoryをpandasで保存して図のplotまで

# 生データの保存
history = model.fit(...)
hist_df = pd.DataFrame(history.history)
hist_df.to_csv('history.csv')

# 可視化
plt.figure()
hist_df[['accuracy', 'val_accuracy']].plot()
plt.ylabel('accuracy')
plt.xlabel('epoch')
plt.savefig(os.path.join(self.result_dir, 'acc.png'))
plt.close()

plt.figure()
hist_df[['loss', 'val_loss']].plot()
plt.ylabel('loss')
plt.xlabel('epoch')
plt.savefig(os.path.join(self.result_dir, 'loss.png'))
plt.close()