首页 > 解决方案 > 如何进行组递归特征消除?

问题描述

我想使用递归特征消除(RFE)进行组特征选择。例如我有如下数据

from sklearn.datasets import make_classification
X,y=make_classification(n_samples=100, n_features=20)

我想要做的是,我不想在单列上应用 RFE,而是在列组中执行。例如,我的前 10 列是一组,10-20 列是第二组,依此类推。所以我想要的是在每次迭代中消除一组而不是一列。因此,如果 Ipassn_features_to_select=5我应该得到 50 列(5 组)而不是 5 列。

estimator = SVR(kernel="linear")
selector = RFE(estimator, n_features_to_select=5)

标签: pythonscikit-learnfeature-selection

解决方案


推荐阅读