使用scikit RFE获取参数数错误

Question

我遇到了这个错误。我认为这是我当地设置的一个问题。

# Importing RFE and LinearRegression
from sklearn.feature_selection import RFE
from sklearn.linear_model import LinearRegression

# Running RFE with the output number of the variable equal to 10
lm = LinearRegression()
lm.fit(X_train, y_train)

rfe = RFE(lm, 10)             # running RFE
rfe = rfe.fit(X_train, y_train)

在jupyter上运行单元格时会出现以下错误

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
~\AppData\Local\Temp/ipykernel_3420/241754350.py in <module>
      3 lm.fit(X_train, y_train)
      4 
----> 5 rfe = RFE(lm, 10)             # running RFE
      6 rfe = rfe.fit(X_train, y_train)

TypeError: __init__() takes 2 positional arguments but 3 were given

上面的代码似乎适用于其他人。

Answer 1

正如您在RFE类的__init__方法(rfe.py#L176)中所看到的，您只能传递两个位置参数(self和estimator)。

根据Python文档

“*”或“*identifier”之后的参数是只使用关键字的参数，只能使用关键字参数传递。

在您的情况下，您应该键入：

rfe = RFE(lm, n_features_to_select=10)