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

[GPU-Plugin] Various fixes (#2579)

Browse Source 
* Fix test large

* Add check for max_depth 0

* Update readme

* Add LBS specialisation for dense data

* Add bst_gpair_precise

* Temporarily disable accuracy tests on test_large.py

* Solve unused variable compiler warning

* Fix max_bin > 1024 error
This commit is contained in:
Rory Mitchell
2017-08-05 22:16:23 +12:00
committed by GitHub
parent 03e213c7cd
commit eda9e180f0
8 changed files with 147 additions and 98 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
plugin/updater_gpu/README.md
View File

@@ -27,7 +27,7 @@ This plugin currently works with the CLI version and python version.
Python example:
```python
param['gpu_id'] = 1
param['gpu_id'] = 0
param['max_bin'] = 16
param['tree_method'] = 'gpu_hist'
```