python - 将 n_features_to_select RFE 设置为管道中的百分比

Question

我有一个像这样的管道:

lin_reg_pipeline = Pipeline([
    ('polynomial_features', PolynomialFeatures()),
    ('normalize_polynomial_features', StandardScaler()),
    ('feature_selection', RFE(LinearRegression(), verbose=1)),
    ('lin_reg', LinearRegression())
])

现在，当在网格搜索中安装此管道时，我指定要调整的以下参数:

params = {
    'polynomial_features__degree': [1, 2, 3],
    'feature_selection__n_features_to_select': st.randint(10, 100)
}

有没有办法可以将 n_features_to_select 设置为数据集中特征总数的百分比？因为我不知道PolynomialFeatures()会添加多少个特征。

提前致谢，

凯文

Answer 1

您可以传递 0.0 到 1.0 之间的 float 作为 step 参数，默认情况下，这将删除每个步骤的百分比。

查看the documentation here