Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
xgboost/demo/gpu_acceleration
History
Rory Mitchell f0c3ff313f
Update GPUTreeShap, add docs (#6281) 
* Update GPUTreeShap, add docs

* Fix test

Co-authored-by: Philip Hyunsu Cho <chohyu01@cs.washington.edu>
2020-10-27 18:22:12 +13:00
..
cover_type.py
Update Python demos with tests. (#5651)
2020-05-12 12:04:42 +08:00
README.md
Update GPUTreeShap, add docs (#6281)
2020-10-27 18:22:12 +13:00
shap.ipynb
Update GPUTreeShap, add docs (#6281)
2020-10-27 18:22:12 +13:00

README.md

GPU Acceleration Demo

cover_type.py shows how to train a model on the forest cover type dataset using GPU acceleration. The forest cover type dataset has 581,012 rows and 54 features, making it time consuming to process. We compare the run-time and accuracy of the GPU and CPU histogram algorithms.

shap.ipynb demonstrates using GPU acceleration to compute SHAP values for feature importance.