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)

Module contents