首页 > 解决方案 > 在 scikit-learn 估计器中保存 X、y 数据

问题描述

在我看来,合适的数据不会保存在估计器上。我正在对课程进行子BaseEstimator类化,并且想知道以下内容:

标签: pythonscikit-learn

解决方案


sklearn 文档中,

请注意,模型是使用 X 和 y 拟合的,但对象不包含对 X 和 y 的引用。然而,这也有一些例外,例如在预计算内核的情况下,必须存储这些数据以供 预测方法使用。

不存储数据的原因是,
一般来说,一旦拟合了估计器,数据本身将不再被使用。

您以后是否会后悔将数据复制到属性取决于多种因素:对应用程序的要求、用例、数据的大小等等。


推荐阅读