Python 如何找到RFECV选择的要素名称?
如何知道使用Python 如何找到RFECV选择的要素名称?,python,pandas,scikit-learn,Python,Pandas,Scikit Learn,如何知道使用RFECV选择了哪些特定功能?如果我做了X\u rfecv\u train=selector.transform(X\u train),我会得到一个numpy数组,但我不知道特性名称 rf_rec = RandomForestClassifier(n_jobs=-1, max_depth = 20, max_features = 0.9, min_samples_leaf = 2, min_samples_split
RFECV
选择了哪些特定功能?如果我做了X\u rfecv\u train=selector.transform(X\u train)
,我会得到一个numpy数组,但我不知道特性名称
rf_rec = RandomForestClassifier(n_jobs=-1, max_depth = 20, max_features = 0.9,
min_samples_leaf = 2, min_samples_split = 0.1,
n_estimators = 100, oob_score=True, class_weight="balanced",
random_state=0)
selector = RFECV(rf_rec, step=1, cv=5)
selector = selector.fit(X_train, y_train)
X_rfecv_train=selector.transform(X_train)
X_rfecv_test=selector.transform(X_test)
您可以使用。scikit learn中的所有选择器都应具有此方法
print(selector.get_support(indices=True))
这将打印您在X\u rfecv\u列车
和X\u rfecv\u测试
中选择和显示的原始功能的索引