reagent.evaluation.feature_importance package
Submodules
reagent.evaluation.feature_importance.feature_importance_base module
- class reagent.evaluation.feature_importance.feature_importance_base.FeatureImportanceBase(model: torch.nn.modules.module.Module, sorted_feature_ids: List[int])
Bases:
object- compute_feature_importance() pandas.core.frame.DataFrame
- model: torch.nn.modules.module.Module
- sorted_feature_ids: List[int]
reagent.evaluation.feature_importance.feature_importance_perturbation module
- class reagent.evaluation.feature_importance.feature_importance_perturbation.FeatureImportancePerturbation(model: torch.nn.modules.module.Module, sorted_feature_ids: List[int], data_loader: Any, pred_fn: Callable[[torch.nn.modules.module.Module, Any], torch.Tensor], perturb_fn: Callable[[Any, int], Any], repeat: int = 1)
Bases:
reagent.evaluation.feature_importance.feature_importance_base.FeatureImportanceBase- compute_feature_importance() pandas.core.frame.DataFrame
- data_loader: Any
- perturb_fn: Callable[[Any, int], Any]
- pred_fn: Callable[[torch.nn.modules.module.Module, Any], torch.Tensor]
- repeat: int = 1
- reagent.evaluation.feature_importance.feature_importance_perturbation.create_default_perturb_fn(key: str)