Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
xgboost/demo/gpu_acceleration/README.md
Rory Mitchell e67388fb8f
Some guidelines on device memory usage (#5038) 
* Add memory usage demo

* Update documentation
2019-11-17 07:48:24 +13:00

456 B
Raw Blame History

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.

memory.py shows how to repeatedly train xgboost models while freeing memory between iterations.