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

[backport] Fix device dispatch for linear updater. (#9507) (#9532)

Browse Source
This commit is contained in:
Jiaming Yuan
2023-08-29 15:10:43 +08:00
committed by GitHub
parent 4301558a57
commit a0d3573c74
6 changed files with 80 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
doc/parameter.rst
View File

@@ -329,7 +329,7 @@ Parameters for Linear Booster (``booster=gblinear``)
- Choice of algorithm to fit linear model
- ``shotgun``: Parallel coordinate descent algorithm based on shotgun algorithm. Uses 'hogwild' parallelism and therefore produces a nondeterministic solution on each run.
- ``coord_descent``: Ordinary coordinate descent algorithm. Also multithreaded but still produces a deterministic solution.
- ``coord_descent``: Ordinary coordinate descent algorithm. Also multithreaded but still produces a deterministic solution. When the ``device`` parameter is set to ``cuda`` or ``gpu``, a GPU variant would be used.
* ``feature_selector`` [default= ``cyclic``]