Decision Trees Python Packages

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.

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.

tensorflow-decision-forests

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

ydf

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

orange3

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

dtreeviz

A python library for decision tree visualization and model interpretation.

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.

lleaves

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

linear-tree

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

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.

adaxt

adaXT: tree-based machine learning in Python

stochtree

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

ydf-tf

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

pynets

A Reproducible Workflow for Structural and Functional Connectome Ensemble Learning

fukuml

Simple machine learning library

pyscipopt-ml

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

gosdt

Generalized Optimal Sparse Decision Trees

supertree

Visualize decision trees in Python

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

dectree

A tool to explore decision trees

spkit

Signal Processing Toolkit, including ML models with visualization

grande

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

infofuzzynetwork

An independent implementation of the IFN model, published on PYPI (available on pip)

trust-free

TRUST - Decision Trees with Sparse Elastic Net leaves, Random Forest accuracy and automated explanations