ray-project / xgboost_ray

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

Compatibility with xgboost 1.4 (remove feature names form numpy arrays) #85

Closed krfricke closed 3 years ago

krfricke commented 3 years ago

Converting numpy arrays to pandas dataframes for sharding implicitly adds coluimn names which are stored in the reuslting xgboost booster objects. However, calling predict on such boosters with numpy arrays leads to errors. Instead we should unset feature names for numpy arrays if they haven't been deliberately passed to RayDMatrix.