获取 TypeError: '(slice(None, None, None), array([0, 1, 2, 3, 4]))' 是一个无效的键。

6

尝试使用BorutaPy进行特征选择,但出现TypeError错误:“(slice(None,None,None),array([0,1,2,3,4]))”是无效的键。

from sklearn.ensemble import RandomForestClassifier
from boruta import BorutaPy
rf = RandomForestClassifier(n_jobs=-1, max_depth=4)

# define Boruta feature selection method
feat_selector = BorutaPy(rf, n_estimators='auto', verbose=2, random_state=1)

X = train_dt[['age', 'menopause', 'tumor_size', 'inv_nodes', 'node_caps',
   'deg_malig', 'breast', 'breast_quad', 'irradiat']]
Y = train_dt.label

# find all relevant features - 5 features should be selected
feat_selector.fit(x, y)

# check selected features - first 5 features are selected
feat_selector.support_

# check ranking of features
feat_selector.ranking_

# call transform() on X to filter it down to selected features
X_filtered = feat_selector.transform(X)

我使用了乳腺癌数据集,并进行了一些小的调整,比如添加标题、特征缩放和缺失值处理。


你在哪一行遇到了错误? - ddaletski
1个回答

5

我也遇到了同样的错误。我能够追溯到pandas的IndexEngine,但无法确定问题出在哪里。您可以通过将数据框转换为numpy数组来运行模型,例如:

feat_selector.fit(x.values, y.values)

此外,你在上一行将xy命名为XY,但我确定如果你实际使用这样的代码,它会显示不同的错误。


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接