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.

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.

dtreeviz

A python library for decision tree visualization and model interpretation.

orange3

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

lleaves

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

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.

mojolearn

GPU machine learning in Mojo that supports bitwise identical inference and training on Apple Metal, NVIDIA CUDA and AMD HIP.

gosdt

Generalized Optimal Sparse Decision Trees

ydf-tf

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

stochtree

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

dectree

A tool to explore and visualize decision trees

prismboost

PrismBoost: gradient boosting with SEFR oblique splits for classification and regression

linear-tree

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

mojolearn-nvidia

GPU machine learning in Mojo that supports bitwise identical inference and training on Apple Metal, NVIDIA CUDA and AMD HIP.

adaxt

adaXT: tree-based machine learning in Python

mojolearn-amd

GPU machine learning in Mojo that supports bitwise identical inference and training on Apple Metal, NVIDIA CUDA and AMD HIP.

c50py

C5.0 decision trees and rulesets for Python: native categorical splits, missing values, boosting and rules exported to SQL, as scikit-learn estimators

neural-trees

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

pynets

A Reproducible Workflow for Structural and Functional Connectome Ensemble Learning

r2py-rpart

Python port of R's rpart: recursive partitioning and regression trees, with a compiled C core (via cffi) and R-parity Python functions for fitting, pruning, predicting, and plotting trees. Supports classification, regression, Poisson/survival, and user-defined splits. GPL-2.0-or-later, derived from Therneau & Atkinson's rpart.

supertree

Impress your boss with interactive Decision Tree visualization