tabensemb.trainer.Trainer.cal_partial_dependence#
method
- Trainer.cal_partial_dependence(feature_subset: List[str] | None = None, **kwargs) Tuple[List[ndarray], List[ndarray], List[ndarray], List[ndarray]][source]#
Calculate partial dependency. See the source code of
plot_partial_dependence()for its usage.- Parameters:
- feature_subset
A subset of
all_feature_names().- kwargs
Arguments for
_bootstrap_fit().
- Returns:
- list
x values for each feature
- list
pdp values for each feature
- list
lower confidence limits for each feature
- list
upper confidence limits for each feature