parrt/dtreeviz
A python library for decision tree visualization and model interpretation.
Supports scikit-learn, XGBoost, LightGBM, Spark MLlib, and TensorFlow Decision Forests through a unified adaptor interface, enabling consistent visualization APIs across frameworks. Generates SVG/PNG visualizations showing tree structure, individual prediction paths with feature contributions, leaf distributions, and decision boundaries for both classification and regression. Also provides feature space exploration tools like decision boundary visualization and hyperparameter animation for model interpretation.
3,128 stars and 98,489 monthly downloads. Available on PyPI.
Stars
3,128
Forks
340
Language
Jupyter Notebook
License
MIT
Category
Last pushed
Jan 02, 2026
Monthly downloads
98,489
Commits (30d)
0
Dependencies
6
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/parrt/dtreeviz"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related frameworks
google/yggdrasil-decision-forests
A library to train, evaluate, interpret, and productionize decision forest models such as Random...
tensorflow/decision-forests
A collection of state-of-the-art algorithms for the training, serving and interpretation of...
neurodata/treeple
Scikit-learn compatible decision trees beyond those offered in scikit-learn
sanity/quickml
A fast and easy to use decision tree learner in java
lagodiuk/decision-tree-js
Small JavaScript implementation of ID3 Decision tree