如何使这个估计量scikitlearncompatible?

2024-10-03 21:35:40 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图使这个估计器scikit learn兼容,这样我就可以用GridSearchCV搜索参数空间。在

编辑:

我已经按照建议修改了脚本(见下文)。在

  • fit签名被修改为fit(self, X, y)
  • 所有参数都传入__init__

仍然存在与GripdSearchCV的兼容性问题,可能是因为估计器是一个多标签分类器。在

ValueError: Can't handle mix of multilabel-indicator and continuous-multioutput

但这超出了重点;属性错误现在已经消失了。因此,我们可以有把握地得出结论,所建议的修改使估计器scikit-learn兼容。在

最终代码脚本:

^{pr2}$

Tags: self脚本编辑参数分类器init空间标签
1条回答
网友
1楼 · 发布于 2024-10-03 21:35:40

get_params方法中调用self.itrs,但对象没有这样的属性。在

另外,我建议您将fit签名改为fit(self, X, y)

  1. 传递__init__中的所有参数
  2. 分离X和{},使用sklearn.cross_validation.train_test_split进行训练和测试。在

这将使你的代码更像sklearn,并且与库函数更加兼容。在

相关问题 更多 >