scikit-learn_bench icon indicating copy to clipboard operation
scikit-learn_bench copied to clipboard

Make use of "--device(s)" for XGBoost

Open Alexsandruss opened this issue 3 years ago • 0 comments

CPU and GPU configs for XGBoost have only few differences: in data-format (pandas vs cudf) and tree-method (hist vs gpu_hist). Dispatching for them with --devices(s) argument will simplify configs.

Alexsandruss avatar Dec 23 '21 14:12 Alexsandruss