search

oneapi-src / oneDAL

oneAPI Data Analytics Library (oneDAL)
https://software.intel.com/en-us/oneapi/onedal
Apache License 2.0
608 stars 211 forks source link

Support for accelerating NGBoost #2132

Open Geethen opened 1 year ago

Geethen commented 1 year ago

I have come across speed-ups for XGBoost and LightGBM here and wondered if it similar sppedups could be achieved for NGBoost with LightGBM as a base learner. https://stanfordmlgroup.github.io/projects/ngboost/

Alexsandruss commented 1 year ago

Thank you for information, NGBoost looks interesting. Currently, oneDAL does not have same patching functionality for LightGBM and XGBoost as for sklearn, but this direction is promising. Theoretically, you can implement estimator inherited from LightGBM and implement custom predict and predict_proba methods using daa4py functions