python - 获取类型错误 : '(slice(None, None, None), array([0, 1, 2, 3, 4]))' is an invalid key

Question

尝试使用 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)

我用过乳腺癌dataset并做了一些小的调整，例如添加 header 、特征缩放和缺失值处理。

Answer 1

我也遇到了同样的错误。我可以追溯到 pandas IndexEngine，但无法弄清楚到底出了什么问题。您可以通过将数据帧转换为 numpy 数组来运行模型，例如:

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

此外，您在之前的行中将 x 和 y 命名为 X 和 Y，但我'我确信如果您实际上使用了这样的代码，那会显示为不同的错误。