在调用sklearn fit时出现 Unknown label type: 'unknown'错误

在调用sklearn selectKBest().fit时出现 Unknown label type: 'unknown'错误:

当出现此问题时一般来说是你传入的数据类型被调用函数不接受, 你需要改变数据类型。

1.

例如采用卡方验证选取特征时,如果你传入的是标签是float类型,会产生此错误。

后加上astype(‘int’)即可 例如:

                                                                                 .fit(train_x,train_y.astype(‘int’))

2.如果你使用的是回归预测标签需要float类型, 如下例 你不可以使用chi2 它的里面不接受浮点数,只接受分类的类标签值(int)可以用 f_regression

详细内容参考:

附网址:http://scikit-learn.org/stable/modules/feature_selection.html#univariate-feature-selection

猜你喜欢

转载自blog.csdn.net/H_hei/article/details/83626955