在 Keras 中,我们可以返回以下输出model.fit
历史记录如下:
history = model.fit(X_train, y_train,
batch_size=batch_size,
nb_epoch=nb_epoch,
validation_data=(X_test, y_test))
现在,如何将历史对象的历史属性保存到文件中以供进一步使用(例如绘制 acc 或 loss 相对于历元的图)?
我使用的是以下内容:
with open('/trainHistoryDict', 'wb') as file_pi:
pickle.dump(history.history, file_pi)
通过这种方式,我将历史记录保存为字典,以防我以后想绘制损失或准确性。稍后,当您想再次加载历史记录时,可以使用:
with open('/trainHistoryDict', "rb") as file_pi:
history = pickle.load(file_pi)
为什么选择pickle而不是json?
下的评论这个答案 https://stackoverflow.com/a/53101097/11659881准确地指出:
[将历史记录存储为 json] 在 Tensorflow keras 中不再起作用。我遇到了以下问题:TypeError:“float32”类型的对象不可 JSON 序列化。
有办法告诉json
如何编码numpy
对象,您可以从中了解其他问题 https://stackoverflow.com/q/26646362/11659881,所以使用没有什么问题json
在这种情况下,它比简单地转储到 pickle 文件更复杂。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)