Exact computation of Shapley R-squared in polynomial time
Project description
## Q-SHAP: Feature-Specific $R^2$ Values for Tree Ensembles This package is used to compute feature-specific $R^2$ values, following Shapley decomposition of the total $R^2$, for tree ensembles in polynomial time.
This version only takes output from XGBoost, scikit-learn Decision Tree, and scikit-learn GBDT. We are working to update it for random forest in the next version. Please check Q-SHAP Tutorial.ipynb for more usage.
### Citation `bibtex @article{jiang2024feature, title={Feature-Specific Coefficients of Determination in Tree Ensembles}, author={Jiang, Zhongli and Zhang, Dabao and Zhang, Min}, journal={arXiv preprint arXiv:2407.03515}, year={2024} } `
### References - Jiang, Z., Zhang, D., & Zhang, M. (2024). Feature-Specific Coefficients of Determination in Tree Ensembles. arXiv preprint arXiv:2407.03515. - Lundberg, Scott M., et al. “From local explanations to global understanding with explainable AI for trees.” Nature machine intelligence 2.1 (2020): 56-67. - Karczmarz, Adam, et al. “Improved feature importance computation for tree models based on the Banzhaf value.” Uncertainty in Artificial Intelligence. PMLR, 2022. - Bifet, Albert, Jesse Read, and Chao Xu. “Linear tree shap.” Advances in Neural Information Processing Systems 35 (2022): 25818-25828. - Chen, Tianqi, and Carlos Guestrin. “Xgboost: A scalable tree boosting system.” Proceedings of the 22nd acm sigkdd international conference on knowledge discovery and data mining. 2016.
### Container images We provide pre-built images with all necessary packages for Q-SHAP in Python 3.11, available for both Docker and Singularity:
Docker: You can pull the Docker image using the following command: `sh docker pull catstat/xai `
Singularity: You can pull the Docker image using the following command: `sh singularity pull docker://catstat/xai:0.0 `
### Task List
[ ] Task 1: Lightgbm version
[ ] Task 2: Catboost version
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file qshap-0.1.1.tar.gz
.
File metadata
- Download URL: qshap-0.1.1.tar.gz
- Upload date:
- Size: 14.4 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.12.4
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | ffb65a1f37ae4352b996ac9af732061bbc33aab29fc856fc0f9cfa7a72e96860 |
|
MD5 | cc9787d3941855dfb0e86dc1d3f7a33e |
|
BLAKE2b-256 | 0828cda24f3f13e9a74bb942d5d0ff482e766c2d97027b35ef7b31e6813164d7 |
File details
Details for the file qshap-0.1.1-py3-none-any.whl
.
File metadata
- Download URL: qshap-0.1.1-py3-none-any.whl
- Upload date:
- Size: 17.3 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/5.1.1 CPython/3.12.4
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | ccfdee5309db6f0892e7fa274c0dfc6b8421a4e9849a656a72efae11f623b28f |
|
MD5 | ead6c275742eac61d32e54e4d26cf97e |
|
BLAKE2b-256 | 81c5a5b4187cdd0847f87dc520591fbf57a96fd4caad744dd7c031bd92f794ae |