powershap is a feature selection method that uses statistical hypothesis testing and power calculations on Shapley values, enabling fast and intuitive wrapper-based feature selection.
Installation βοΈ
| pip | pip install powershap |
|---|
Usage π
powershap is built to be intuitive, it supports various models including linear, tree-based, and even deep learning models for classification and regression tasks.
from powershap import PowerShap
from catboost import CatBoostClassifier
X, y = ... # your classification dataset
selector = PowerShap(
model=CatBoostClassifier(n_estimators=250, verbose=0, use_best_model=True)
)
selector.fit(X, y) # Fit the PowerShap feature selector
selector.transform(X) # Reduce the dataset to the selected featuresFeatures β¨
- default automatic mode
scikit-learncompatible- supports various models
- insights into the feature selection method: call the
._processed_shaps_dfon a fittedPowerSHAPfeature selector. - tested code!
Benchmarks β±
Check out our benchmark results here.
How does it work βοΈ
Powershap is built on the core assumption that an informative feature will have a larger impact on the prediction compared to a known random feature.
- Powershap trains multiple models with different random seeds on different subsets of the data. Each iteration it adds a random uniform feature to the dataset for training.
- In a single iteration after training a model, powershap calculates the absolute Shapley values of all features, including the random feature. If there are multiple outputs or multiple classes, powershap uses the maximum across these multiple outputs. These values are then averaged for each feature, symbolising the impact of the feature in this iteration.
- After performing all iterations, each feature then has an array of impacts. The impact array of each feature is then compared to the average of the random feature impact array using the percentile formula to provide a p-value. This tests whether the feature has a larger impact than the random feature and outputs a low p-value if true.
- Powershap then outputs all features with a p-value below the provided threshold. The threshold is by default 0.01.
Automatic mode π€
The required number of iterations and the threshold values are hyperparameters of powershap. However, to avoid manually optimizing the hyperparameters powershap by default uses an automatic mode that automatically determines these hyperparameters.
- The automatic mode first starts with executing powershap using ten iterations.
- Then, for each feature powershap calculates the effect size and the statistical power of the test using a student-t power test.
- Using the calculated effect size, powershap then calculates the required iterations to achieve a predefined power requirement. By default this is 0.99, which represents a false positive probability of 0.01.
- If the required iterations are larger than the already performed iterations, powershap then further executes for the extra required iterations.
- Afterward, powershap re-calculates the required iterations and it keeps re-executing until the required iterations are met.
Referencing our package π
If you use powershap in a scientific publication, we would highly appreciate citing us as:
@InProceedings{10.1007/978-3-031-26387-3_5,
author="Verhaeghe, Jarne
and Van Der Donckt, Jeroen
and Ongenae, Femke
and Van Hoecke, Sofie",
title="Powershap: A Power-Full Shapley Feature Selection Method",
booktitle="Machine Learning and Knowledge Discovery in Databases",
year="2023",
publisher="Springer International Publishing",
address="Cham",
pages="71--87",
isbn="978-3-031-26387-3"
}Paper was presented at ECML PKDD 2022. The manuscript can be found here and on the github.
License
This package is available under the MIT license. More information can be found here: https://github.com/predict-idlab/powershap/blob/main/LICENSE