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 之间的浮点数作为step参数，默认情况下每一步都会删除一个百分比。

在此处查看文档