sklearn管道+ keras顺序模型-如何获取历史记录?

时间:2019-01-25 04:13:32

标签: python-3.x keras scikit-learn pipeline

Keras模型,当调用.fit时,返回一个历史对象。如果我将此模型用作sklearn管道的一个步骤,是否可以检索它? 顺便说一句,我正在使用python 3.6

谢谢!

1 个答案:

答案 0 :(得分:2)

History回调记录每个时期的训练指标。这包括损失和准确性(针对分类问题)以及验证数据集的损失和准确性(如果已设置)。

历史记录对象从对用于训练模型的fit()函数的调用中返回。指标存储在返回对象的历史记录成员中的字典中。

这也意味着这些值必须在fit()函数或顺序模型的范围内,因此,如果它在sklearn管道中,则无法访问最终值,并且无法存储或返回看不到的东西。

截至目前,我还不知道sklearn中的历史记录回调,因此对您来说,我唯一看到的是手动记录要跟踪的指标。一种方法是让管道返回数据,然后简单地将模型拟合到该数据上。如果您无法找出注释。