File: README.md

package info (click to toggle)
xgboost 1.7.4-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 11,432 kB
  • sloc: cpp: 47,464; python: 26,479; java: 3,868; xml: 1,236; sh: 1,095; ansic: 919; makefile: 323; javascript: 19
file content (5 lines) | stat: -rw-r--r-- 454 bytes parent folder | download 
1
2
3
4
5
 
# GPU Acceleration Demo

`cover_type.py` shows how to train a model on the [forest cover type](https://archive.ics.uci.edu/ml/datasets/covertype) 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.