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

时间:2019-11-13 19:30:54

标签: keras callback history

我使用keras回调ModelCheckpoint保存我最好的火车模型。是否可以用相同的方式保存相应的历史记录以进行打印?

1 个答案:

答案 0 :(得分:1)

您可以在回调列表中添加来自keras的https://ufile.io/3iw14o82回调以及ModelCheckpoint。

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

这会将每个时期的所有度量标准信息都转储到CSV文件中,您可以使用该文件进行绘图。 另外, model.fit 返回keras历史记录对象,该对象可进一步用于CSVLogger