scikit-learn_bench
scikit-learn_bench copied to clipboard
Make use of "--device(s)" for XGBoost
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.