keras中如何保存训练历史对象的历史属性?
0 1490
0

在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或损失图)?

收藏
2021-02-05 11:34 更新 karry •  4552
共 1 个回答
高赞 时间
0

我使用的是以下代码:

with open('/trainHistoryDict', 'wb') as file_pi:
        pickle.dump(history.history, file_pi)

这样,我将历史保存为dictionary,以便稍后我想绘制损失或准确性图像。

Via:https://stackoverflow.com/a/44674337/14964791

收藏
2021-02-05 11:59 更新 anna •  5050