8000 Updates SelectFromModel docstring to detail supported estimators (#9238) · scikit-learn/scikit-learn@c5c51c6 · GitHub
[go: up one dir, main page]

Skip to content

Commit c5c51c6

Browse files
jrbourbeaujnothman
authored andcommitted
Updates SelectFromModel docstring to detail supported estimators (#9238)
* Updates SelectFromModel docstring * Updates docstring
1 parent fbcbc1b commit c5c51c6

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

sklearn/feature_selection/from_model.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,8 @@ class SelectFromModel(BaseEstimator, SelectorMixin, MetaEstimatorMixin):
8787
estimator : object
8888
The base estimator from which the transformer is built.
8989
This can be both a fitted (if ``prefit`` is set to True)
90-
or a non-fitted estimator.
90+
or a non-fitted estimator. The estimator must have either a
91+
``feature_importances_`` or ``coef_`` attribute after fitting.
9192
9293
threshold : string, float, optional default None
9394
The threshold value to use for feature selection. Features whose

0 commit comments

Comments
 (0)
0