keras:如何保存历史对象的训练历史属性

jingweimo :

在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] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章