Skip to content

Commit 51fae9f

Browse files
authored
DOC Fix example Recursive feature elimination with cross-validation (scikit-learn#31516)
1 parent dab0842 commit 51fae9f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/feature_selection/plot_rfe_with_cross_validation.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,7 @@
105105

106106
for i in range(cv.n_splits):
107107
mask = rfecv.cv_results_[f"split{i}_support"][
108-
rfecv.n_features_
108+
rfecv.n_features_ - 1
109109
] # mask of features selected by the RFE
110110
features_selected = np.ma.compressed(np.ma.masked_array(feat_names, mask=1 - mask))
111111
print(f"Features selected in fold {i}: {features_selected}")

0 commit comments

Comments
 (0)