Skip to content

Commit

Permalink
Add callbacks docstring in rfe module
Browse files Browse the repository at this point in the history
  • Loading branch information
ClaudioSalvatoreArcidiacono committed Jul 30, 2024
1 parent ba0f5b6 commit 76a970b
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions felimination/rfe.py
Original file line number Diff line number Diff line change
Expand Up @@ -158,6 +158,10 @@ class FeliminationRFECV(RFE):
best) features are assigned rank 1.
support_ : ndarray of shape (n_features,)
The mask of selected features.
callbacks : list of callable, default=None
List of callables to be called at the end of each step of the feature
selection. Each callable should accept two parameters: the selector
and the importances computed at that step.
Examples
--------
Expand Down Expand Up @@ -567,6 +571,10 @@ class PermutationImportanceRFECV(FeliminationRFECV):
it keeps the method tractable when evaluating feature importance on
large datasets. In combination with `n_repeats`, this allows to control
the computational speed vs statistical accuracy trade-off of this method.
callbacks : list of callable, default=None
List of callables to be called at the end of each step of the feature
selection. Each callable should accept two parameters: the selector
and the importances computed at that step.
Attributes
Expand Down

0 comments on commit 76a970b

Please sign in to comment.