sklearn rfecv的功能选择结果不一致

时间:2019-01-28 08:52:45

标签: python scikit-learn

我试图通过RFECV进行功能选择:

X_train, X_test, y_train, y_test = train_test_split(X_matrix,
                                                    y, test_size=0.2, random_state=42)

selector = RFECV(DecisionTreeRegressor(), min_features_to_select=5,
                  step=5, cv=2, n_jobs=-1)
selector.fit(X_train ,y_train)


print(selector.support_)

打印的输出是用于功能选择的遮罩,但是每次执行都会返回一个不同的遮罩。 X和y数据框都可以。

输出:

1

[True True True True True True False False False True正确False True   真真真真假真真真假真真真真   真真真真真真真假真真真真真   True True True True True True False False True True False True True]

2

[True True True True True True True True True True True True True True   真真真真真真真真真真真真真真真   真真真真真真真真真真真真真真真   True True True True True True True True True True True True True]

3

[False True True False False True True False False False True True False False  错误错误错误错误错误错误错误错误错误错误错误  错误错误正确正确错误False False False False False False False False True   正确正确正确正确错误False False False False False False False]

以此类推...

可能是什么原因?

0 个答案:

没有答案