PyRank
  • Insights
  • PyPI
  • GitHub
  • Search
  • Compare
  • Advisories
  • Ecosystem
  • About

Gradient Boosting Python Packages

Python packages with the GitHub topic gradient-boosting. Sorted by relevance, with stars and monthly downloads.
microsoft
lightgbm

A fast, distributed, high performance gradient boosting (GBT, GBDT, GBRT, GBM or MART) framework based on decision tree algorithms, used for ranking, classification and many other machine learning tasks.

16.2M 18K 4K
shap
shap

A game theoretic approach to explain the output of any machine learning model.

14.8M 25K 4K
catboost
catboost

A fast, scalable, high performance Gradient Boosting on Decision Trees library, used for ranking, classification, regression and other machine learning tasks for Python, R, Java, C++. Supports computation on CPU and GPU.

6.4M 9K 1K
interpretml
interpret-core

Fit interpretable models. Explain blackbox machine learning.

944K 7K 784
tensorflow
tensorflow-decision-forests

A collection of state-of-the-art algorithms for the training, serving and interpretation of Decision Forest models in Keras.

677K 692 116
google
ydf

A library to train, evaluate, interpret, and productionize decision forest models such as Random Forest and Gradient Boosted Decision Trees.

557K 656 80
interpretml
interpret

Fit interpretable models. Explain blackbox machine learning.

429K 7K 784
ottenbreit-data-science
aplr

APLR builds predictive, interpretable regression and classification models using Automatic Piecewise Linear Regression. It often rivals tree-based methods in predictive accuracy while offering smoother and interpretable predictions.

224K 23 5
stanfordmlgroup
ngboost

Natural Gradient Boosting for Probabilistic Prediction

176K 2K 249
catboost
catboost-dev

A fast, scalable, high performance Gradient Boosting on Decision Trees library, used for ranking, classification, regression and other machine learning tasks for Python, R, Java, C++. Supports computation on CPU and GPU.

49K 9K 1K
EpistasisLab
tpot

A Python Automated Machine Learning tool that optimizes machine learning pipelines using genetic programming.

31K 10K 2K
siboehm
lleaves

Compiler for LightGBM gradient-boosted trees, based on LLVM. Speeds up prediction by ≥10x.

24K 471 45
aertslab
arboreto

A scalable python-based framework for gene regulatory network inference using tree-based ensemble regressors.

13K 67 37
wsperat
forestfire-ml

Machine learning from the ground up in Rust

9K 0 0
mrapp-ke
mlrl-boomer

A scikit-learn implementation of BOOMER - An Algorithm for Learning Gradient Boosted Multi-Output Rules

8K 5 2
mrapp-ke
mlrl-common

A scikit-learn implementation of BOOMER - An Algorithm for Learning Gradient Boosted Multi-Output Rules

8K 5 2
mrapp-ke
mlrl-seco

A scikit-learn implementation of BOOMER - An Algorithm for Learning Gradient Boosted Multi-Output Rules

7K 5 2
erdogant
treeplot

Plot tree based machine learning models

6K 11 2
ClimbsRocks
auto-ml

Automated machine learning for production and analytics

4K 2K 309
RektPunk
mqboost

Multiple quantiles estimation model maintaining non-crossing condition (or monotone quantile condition) and preventing quantile crossing using LightGBM and XGBoost

4K 41 8
google
ydf-tf

A library to train, evaluate, interpret, and productionize decision forest models such as Random Forest and Gradient Boosted Decision Trees.

3K 656 80
TorchEnsemble-Community
torchensemble

A unified ensemble framework for PyTorch to improve the performance and robustness of your deep learning model.

2K 1K 95
serengil
chefboost

A Lightweight Decision Tree Framework supporting regular algorithms: ID3, C4.5, CART, CHAID and Regression Trees; some advanced techniques: Gradient Boosting, Random Forest and Adaboost w/categorical features support for Python

2K 486 102
ddbourgin
numpy-ml

Machine learning, in numpy

2K 16K 4K
    • Data from PyPI, GitHub, ClickHouse, and BigQuery