How to enable GPU on GradientBoostingClassifier?
Is there a way to enable GPU on GradientBoostingClassifier?
Topic natural-gradient-boosting gpu scikit-learn python
Category Data Science
Per sklearn docs the answer is NO:
Will you add GPU support?
No, or at least not in the near future. The main reason is that GPU support will introduce many software dependencies and introduce platform specific issues. scikit-learn is designed to be easy to install on a wide variety of platforms. Outside of neural networks, GPUs don’t play a large role in machine learning today, and much larger gains in speed can often be achieved by a careful choice of algorithms.
However, xgboost, LightGBM or Catboost do support GPU (although it sometimes is hard to get the GPU runnig at least from what I tried).
When using sklearn, a relatively fast way to train sklearn.ensemble.HistGradientBoostingClassifier. It is way faster than the "normal" GradientBoostingClassifier.
About
Geeks Mental is a community that publishes articles and tutorials about Web, Android, Data Science, new techniques and Linux security.