Accelerating the XGBoost algorithm using GPU computing
Published in PeerJ Computer Science • Jul 24, 2017
Authors:,
Rory Mitchell
Eibe Frank
Abstract
We present a CUDA-based implementation of a decision tree construction algorithm within the gradient boosting library XGBoost. The tree construction algorithm is executed entirely on the graphics processing unit (GPU) and shows high performance with a variety of datasets and settings, including spar...
Finding related papers...
Discussions
(0)No comments yet
Be the first to share your thoughts!