search

ray-project / xgboost_ray

Distributed XGBoost on Ray
Apache License 2.0
133 stars 34 forks source link

Add QuantileDMatrix support #279

Closed kira-lin closed 1 year ago

kira-lin commented 1 year ago

xgboost 1.7.0 has extended quantile dmatrix to CPU, and unified the API. Now DeviceQuantileDMatrix is deprecated. This PR adds support for QuantileDMatrix by adding a thin wrapper.