在回调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历史记录对象，该对象可进一步用于visualization。