A python library for decision tree visualization and model interpretation. Decision trees are the fundamental building block of gradient boosting machines and Random Forests(tm), probably the two most popular machine learning models for structured data. Visualizing decision trees is a tremendous aid when learning how these models work and when interpreting models. The visualizations are inspired by an educational animation by R2D3; A visual introduction to machine learning. Please see How to visualize decision trees for deeper discussion of our decision tree visualization library and the visual design decisions we made.
Features
- Supports scikit-learn, XGBoost, Spark MLlib, LightGBM, and Tensorflow. See Installation instructions
- Sample Visualizations
- Prediction path explanations
- Leaf information
- Documentation available
- Verify graphviz installation
- Install dtreeviz locally
Categories
Machine LearningLicense
MIT LicenseFollow dtreeviz
Other Useful Business Software
Airlock Digital - Application Control (Allowlisting) Made Simple
For organizations seeking the most effective way to prevent malware and ransomware in their environments. It has been designed to provide scalable, efficient endpoint security for organizations with even the most diverse architectures and rigorous compliance requirements. Built by practitioners for the world’s largest and most secure organizations, Airlock Digital delivers precision Application Control & Allowlisting for the modern enterprise.
Rate This Project
Login To Rate This Project
User Reviews
Be the first to post a review of dtreeviz!