Random Forest Python Packages

flaml

A fast library for AutoML and tuning. Join our Discord: https://discord.gg/Cppx2vSPVP.

h2o

H2O is an Open Source, Distributed, Fast & Scalable Machine Learning Platform: Deep Learning, Gradient Boosting (GBM) & XGBoost, Random Forest, Generalized Linear Modeling (GLM with Elastic Net), K-Means, PCA, Generalized Additive Models (GAM), RuleFit, Support Vector Machine (SVM), Stacked Ensembles, Automatic Machine Learning (AutoML), etc.

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.

quantile-forest

Quantile Regression Forests compatible with scikit-learn.

dtreeviz

A python library for decision tree visualization and model interpretation.

orange3

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

miceforest

Multiple Imputation with LightGBM in Python

smac

SMAC3: A Versatile Bayesian Optimization Package for Hyperparameter Optimization

fasttreeshap

Fast SHAP value computation for interpreting tree-based models

tpot

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

emlearn

Machine Learning inference engine for Microcontrollers and Embedded devices

ydf-tf

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

mljar-supervised

Python package for AutoML on Tabular Data with Feature Engineering, Hyper-Parameters Tuning, Explanations and Automatic Documentation

pyxai

PyXAI (Python eXplainable AI) is a Python library (version 3.6 or later) allowing to bring formal explanations suited to (regression or classification) tree-based ML models (Decision Trees, Random Forests, Boosted Trees, ...).

linear-tree

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

arboreto

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

pyabcranger

ABC random forests for model choice and parameter estimation, pure C++ implementation

skranger

scikit-learn compatible Python bindings for ranger C++ random forest library

cve-riskengine-rf

Leakage-free evaluation of a Recursive Feature Elimination Random Forest for CVE exploitability prioritization: code, data, results and a calibrated scoring package.

comprisk

Scalable, scikit-learn-compatible competing-risks survival analysis in pure Python — CR random survival forest, Fine-Gray, cause-specific Cox, Aalen-Johansen CIF, Gray's test, and exact TreeSHAP. 10–22× faster than randomForestSRC on real EHR and 16.6–544× vs scikit-survival (n=5k→50k); scales to n=10⁶ in ~1 min.

h2o-client

H2O is an Open Source, Distributed, Fast & Scalable Machine Learning Platform: Deep Learning, Gradient Boosting (GBM) & XGBoost, Random Forest, Generalized Linear Modeling (GLM with Elastic Net), K-Means, PCA, Generalized Additive Models (GAM), RuleFit, Support Vector Machine (SVM), Stacked Ensembles, Automatic Machine Learning (AutoML), etc.

supertree

Impress your boss with interactive Decision Tree visualization

deep-forest

An Efficient, Scalable and Optimized Python Framework for Deep Forest (2021.2.1)