oneDAL icon indicating copy to clipboard operation
oneDAL copied to clipboard

Support for accelerating NGBoost

Open Geethen opened this issue 3 years ago • 1 comments

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/

Geethen avatar Oct 14 '22 11:10 Geethen

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

Alexsandruss avatar Oct 14 '22 15:10 Alexsandruss