search

oap-project / raydp

RayDP provides simple APIs for running Spark on Ray and integrating Spark with AI libraries.
Apache License 2.0
293 stars 66 forks source link

Add an estimator API for XGBoost using Ray's XGBoostTrainer #289

Closed kira-lin closed 1 year ago

kira-lin commented 1 year ago

We now have 2 estimators, for tensorflow and pytorch. This PR adds another one for xgboost. The XGBoostTrianer's interface is already simple, and our estimator adds an function to work on spark dataframes.