SKlearn Pipeline + Keras Sequential Model - 如何获取历史记录



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

提前感谢!

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

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

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

截至目前,我不知道 sklearn 中有历史回调,所以我唯一看到的是手动记录您要跟踪的指标。一种方法是让管道返回数据,然后简单地将模型拟合到数据上。如果您无法弄清楚,请发表评论。

相关内容

  • 没有找到相关文章

最新更新