在回调 ModelCheckpoint 上使用模型保存历史记录

Question

我使用 keras 回调 ModelCheckpoint 来保存我最好的火车模型。是否可以以相同的方式保存相应的历史以进行绘图？

Answer 1

您可以在回调列表中添加来自 keras 的CSVLogger回调以及 ModelCheckpoint。

history = model.fit(X,y, epochs=100, callbacks=[keras.callbacks.CSVLogger('history.csv')])

这会将每个时期的所有指标信息转储到 CSV 文件，您可以使用该文件进行绘图。此外，model.fit返回keras History对象，该对象可以进一步用于可视化。