logml.feature_importance.extractors.model_based

Classes

ModelBasedImportanceExtractor([config])

Simply fits models using a given dataset and extracts inner feature importance from the fitted model, other "coef_" or "feature_importances" attributes.

class logml.feature_importance.extractors.model_based.ModelBasedImportanceExtractor(config: Optional[dict] = None, **kwargs)

Bases: logml.feature_importance.base.BaseImportanceExtractor

Simply fits models using a given dataset and extracts inner feature importance from the fitted model, other “coef_” or “feature_importances” attributes.

LABEL = 'model_based'

extract_model_feature_importance(model_name: Optional[str] = None, model_cls: Optional[Type[logml.models.base.BaseModel]] = None, params: Optional[dict] = None, dataset: Optional[logml.data.datasets.cv_dataset.ModelingDataset] = None, model=None)

Feature importance extraction for single model.

Parameters

model –

raw_fis: Dict[str, List]