PyPI Stats
  • Insights
  • PyPI
  • GitHub
  • Search
  • Compare
  • Advisories
  • Ecosystem
  • About
Home

Decision Trees Python Packages

Python packages with the GitHub topic decision-trees. 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.

15.3M 18K 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
tensorflow
tensorflow-decision-forests

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

649K 693 116
google
ydf

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

537K 652 78
biolab
orange3

🍊 :bar_chart: :bulb: Orange: Interactive data analysis

100K 6K 1K
parrt
dtreeviz

A python library for decision tree visualization and model interpretation.

98K 3K 340
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.

45K 9K 1K
cerlymarco
linear-tree

A python library to build Model Trees with Linear Models at the leaves.

18K 389 61
siboehm
lleaves

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

17K 469 44
NiklasPfister
adaxt

adaXT: tree-based machine learning in Python

10K 11 2
RGF-team
rgf-python

Home repository for the Regularized Greedy Forest (RGF) library. It includes original implementation from the paper and multithreaded one written in C++, along with various language-specific wrappers.

9K 383 58
StochasticTree
stochtree

Stochastic tree ensembles (BART / XBART) for supervised learning and causal inference

8K 75 17
dPys
pynets

A Reproducible Workflow for Structural and Functional Connectome Ensemble Learning

5K 125 41
ubc-systopia
gosdt

Generalized Optimal Sparse Decision Trees

4K 62 22
google
ydf-tf

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

3K 652 78
fukuball
fukuml

Simple machine learning library / 簡單易用的機器學習套件

2K 284 71
mljar
supertree

Visualize decision trees in Python

2K 532 18
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
Opt-Mucca
pyscipopt-ml

Python interface to automatically formulate Machine Learning models into Mixed-Integer Programs

2K 42 3
Nikeshbajaj
spkit

Signal Processing Toolkit, including ML models with visualization

1K 37 33
Dessia-tech
dectree

A tool to explore and visualize decision trees

1K 2 1
cgrtml
neural-trees

sklearn-compatible PyTorch implementations of Soft Decision Trees, HMoE, and classifier comparison tests (5×2cv F-test). pip install neural-trees

843 23 0
s-marton
grande

(ICLR 2024) GRANDE: Gradient-Based Decision Tree Ensembles

741 101 12
kossisoroyce
timber-compiler

Ollama for classical ML models. AOT compiler that turns XGBoost, LightGBM, scikit-learn, CatBoost & ONNX models into native C99 inference code. One command to load, one command to serve. 336x faster than Python inference.

630 677 20
    • Data from PyPI, GitHub, ClickHouse, and BigQuery