Keras模型,当调用.fit
时,返回一个历史对象。如果我将此模型用作sklearn管道的一个步骤,是否可以检索它?
顺便说一句,我正在使用python 3.6
谢谢!
答案 0 :(得分:2)
History回调记录每个时期的训练指标。这包括损失和准确性(针对分类问题)以及验证数据集的损失和准确性(如果已设置)。
历史记录对象从对用于训练模型的fit()
函数的调用中返回。指标存储在返回对象的历史记录成员中的字典中。
这也意味着这些值必须在fit()
函数或顺序模型的范围内,因此,如果它在sklearn管道中,则无法访问最终值,并且无法存储或返回看不到的东西。
截至目前,我还不知道sklearn中的历史记录回调,因此对您来说,我唯一看到的是手动记录要跟踪的指标。一种方法是让管道返回数据,然后简单地将模型拟合到该数据上。如果您无法找出注释。