我试图使这个估计器scikit learn兼容,这样我就可以用GridSearchCV搜索参数空间。在
编辑:
我已经按照建议修改了脚本(见下文)。在
fit(self, X, y)
__init__
仍然存在与GripdSearchCV的兼容性问题,可能是因为估计器是一个多标签分类器。在
ValueError: Can't handle mix of multilabel-indicator and continuous-multioutput
但这超出了重点;属性错误现在已经消失了。因此,我们可以有把握地得出结论,所建议的修改使估计器scikit-learn兼容。在
最终代码脚本:
^{pr2}$
在
get_params
方法中调用self.itrs
,但对象没有这样的属性。在另外,我建议您将
fit
签名改为fit(self, X, y)
和__init__
中的所有参数X
和{sklearn.cross_validation.train_test_split
进行训练和测试。在这将使你的代码更像sklearn,并且与库函数更加兼容。在
相关问题 更多 >
编程相关推荐