Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

GPU Plugin: Add subsample, colsample_bytree, colsample_bylevel (#1895)

Browse Source
This commit is contained in:
Rory Mitchell
2016-12-23 04:30:36 +13:00
committed by Tianqi Chen
parent cee4aafb93
commit b49b339183
10 changed files with 331 additions and 324 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
plugin/updater_gpu/README.md
View File

@@ -32,8 +32,6 @@ Data is stored in a sparse format. For example, missing values produced by one h
A 4GB graphics card will process approximately 3.5 million rows of the well known Kaggle higgs dataset.
The algorithm will automatically perform row subsampling if it detects there is not enough memory on the device.
## Dependencies
A CUDA capable GPU with at least compute capability >= 3.5 (the algorithm depends on shuffle and vote instructions introduced in Kepler).