ModelBasedImportanceEstimator
- class lightautoml.pipelines.selection.importance_based.ModelBasedImportanceEstimator[source]
Bases:
ImportanceEstimator
Base class for performing feature selection using model feature importances.
- fit(train_valid=None, ml_algo=None, preds=None)[source]
Find the importances of features.
- Parameters:
train_valid (
Optional
[TrainValidIterator
]) – dataset iterator.ml_algo (
Optional
[TypeVar
(ImportanceEstimatedAlgo
, bound=ImportanceEstimator
)]) – ML algorithm used for importance estimation.preds (
Optional
[LAMLDataset
]) – predicted target values.