search

ja-thomas / autoxgboost

autoxgboost - Automatic tuning and fitting of xgboost
Other
122 stars 19 forks source link

Add GPU support #39

Open ja-thomas opened 6 years ago

ja-thomas commented 6 years ago

xgboost also supports using the GPU for training, not sure how much this is actually used, but it should be simple to allow the user to choose

navdeep-G commented 6 years ago

This is not hard to setup. I can help, if needed.

ja-thomas commented 6 years ago

Help is always appreciated, feel free to set up a PR