在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或针对历时的损失图)?
我使用的是以下内容:
with open('/trainHistoryDict', 'wb') as file_pi:
pickle.dump(history.history, file_pi)
这样,我便将历史记录另存为字典,以备日后绘制损失或准确性时使用。
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句