Keras 模型在调用 .fit
时返回一个历史对象。如果我将此模型用作 sklearn 管道的一个步骤,是否可以检索它?顺便说一句,我正在使用Python 3.6
提前感谢!
历史回调记录每个纪元的训练指标。这包括损失和准确性(对于分类问题)以及验证数据集的损失和准确性(如果已设置)。
历史对象是从对用于训练模型的 fit()
函数的调用中返回的。指标存储在返回对象的历史记录成员的字典中。
这也意味着值必须在fit()
函数或顺序模型的范围内,因此如果它在 sklearn 管道中,它无法访问最终值,并且无法存储或返回它看不到的内容。
截至目前,我不知道 sklearn 中有历史回调,所以我唯一看到的是手动记录您要跟踪的指标。一种方法是让管道返回数据,然后简单地将模型拟合到数据上。如果您无法弄清楚,请发表评论。