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

Project Samples

Project Activity

See All Activity >

Categories

Machine Learning

License

MIT License

Follow dtreeviz

dtreeviz Web Site

Other Useful Business Software
Build Securely on Azure with Proven Frameworks Icon
Build Securely on Azure with Proven Frameworks

Lay a foundation for success with Tested Reference Architectures developed by Fortinet’s experts. Learn more in this white paper.

Moving to the cloud brings new challenges. How can you manage a larger attack surface while ensuring great network performance? Turn to Fortinet’s Tested Reference Architectures, blueprints for designing and securing cloud environments built by cybersecurity experts. Learn more and explore use cases in this white paper.
Download Now
Rate This Project
Login To Rate This Project

User Reviews

Be the first to post a review of dtreeviz!

Additional Project Details

Operating Systems

Linux, Mac, Windows

Registered

2024-08-06