Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Library for gradient boosting tree

Do you know of a good library for gradient boosting tree machine learning?

preferably:

  • with good algorithms such as AdaBoost, TreeBoost, AnyBoost, LogitBoost, etc
  • with configurable weak classifiers
  • capable of both classification and prediction (regression)
  • with all kinds of allowed signals: numbers, categories or free text
  • C/C++ or Python
  • opensource

So far I have found http://www.multiboost.org/home which looks good. But I wonder if there are other libraries?

like image 322
Michal Illich Avatar asked Jan 27 '12 15:01

Michal Illich


People also ask

What is gradient boosting library?

Gradient boosting refers to a class of ensemble machine learning algorithms that can be used for classification or regression predictive modeling problems. Gradient boosting is also known as gradient tree boosting, stochastic gradient boosting (an extension), and gradient boosting machines, or GBM for short.

Is CatBoost gradient boosting?

CatBoost is an algorithm for gradient boosting on decision trees. It is developed by Yandex researchers and engineers, and is used for search, recommendation systems, personal assistant, self-driving cars, weather prediction and many other tasks at Yandex and in other companies, including CERN, Cloudflare, Careem taxi.


2 Answers

If you're looking for a python version, the latest release of scikit-learn features gradient boosted regression trees for classification and regression (docs).

It is similar to R's gbm package - gbm is faster for (least-squares) regression wheres scikit-learn's implementation is faster at test-time and when your number of features > 1000.

like image 126
Peter Prettenhofer Avatar answered Oct 22 '22 16:10

Peter Prettenhofer


These don't neccessarily meet all your preferences, but there's also:

  • Treenet a commercialization and extension of Jerome Friedman's original implementation. Not open source but we've found it to work pretty well
  • R gbm package for gradient boosted trees specifically.
like image 24
scttl Avatar answered Oct 22 '22 16:10

scttl