oneDAL icon indicating copy to clipboard operation
oneDAL copied to clipboard

Implement HistGradientBoostingRegressor

Open deadsoul44 opened this issue 3 years ago • 1 comments

I would be nice to speed up the training of this, since it is widely used.

https://scikit-learn.org/stable/modules/generated/sklearn.ensemble.HistGradientBoostingRegressor.html?highlight=histgr#sklearn.ensemble.HistGradientBoostingRegressor

deadsoul44 avatar Mar 23 '22 09:03 deadsoul44

Acceleration of sklearn.ensemble.HistGradientBoosting is not in nearest plans. Existing workloads and code samples show that Gradient Boostings from XGBoost/LightGBM/CatBoost are more popular than scikit-learn's one and we contributed in their performance directly or providing accelerated inference (model builders). However, some kernels inside HistGradientBoosting might be accelerated in future directly or indirectly through acceleration of some common sklearn functions.

Alexsandruss avatar Mar 31 '22 08:03 Alexsandruss