Skip to main content

A Python 3 library for sci-kit learn, XGBoost, LightGBM, Spark, and TensorFlow decision tree visualization

Project description

A python library for decision tree visualization and model interpretation. Decision trees are the fundamental building block of [gradient boosting machines](http://explained.ai/gradient-boosting/index.html) and [Random Forests](https://en.wikipedia.org/wiki/Random_forest)(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](http://www.r2d3.us/); [A visual introduction to machine learning](http://www.r2d3.us/visual-intro-to-machine-learning-part-1/). Please see [How to visualize decision trees](http://explained.ai/decision-tree-viz/index.html) for deeper discussion of our decision tree visualization library and the visual design decisions we made.

Currently dtreeviz supports: [scikit-learn](https://scikit-learn.org/stable), [XGBoost](https://xgboost.readthedocs.io/en/latest), [Spark MLlib](https://spark.apache.org/mllib/), [LightGBM](https://lightgbm.readthedocs.io/en/latest/), and [Tensorflow](https://www.tensorflow.org/decision_forests). See [Installation instructions](README.md#Installation).

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

dtreeviz-2.2.2.tar.gz (77.0 kB view details)

Uploaded Source

Built Distribution

dtreeviz-2.2.2-py3-none-any.whl (91.8 kB view details)

Uploaded Python 3

File details

Details for the file dtreeviz-2.2.2.tar.gz.

File metadata

  • Download URL: dtreeviz-2.2.2.tar.gz
  • Upload date:
  • Size: 77.0 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.2 CPython/3.9.12

File hashes

Hashes for dtreeviz-2.2.2.tar.gz
Algorithm Hash digest
SHA256 a145ff91ffaa11bd2f59186fdcffaa384d90c1757517cbb27e66cc83922319ba
MD5 97b524caaefb1b2d7fe29291c8fd45c2
BLAKE2b-256 097040d3b2ee1a4d68c0cb5c97b23291ce856b9092a097559a3f68504dbed777

See more details on using hashes here.

File details

Details for the file dtreeviz-2.2.2-py3-none-any.whl.

File metadata

  • Download URL: dtreeviz-2.2.2-py3-none-any.whl
  • Upload date:
  • Size: 91.8 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.2 CPython/3.9.12

File hashes

Hashes for dtreeviz-2.2.2-py3-none-any.whl
Algorithm Hash digest
SHA256 e510d80857b29f15a12e8ebe2c7ab3fb667096739742ab47e5d8889cf78da8fe
MD5 c6cdd4bc50640bd88f43b91b82afa919
BLAKE2b-256 b3e844eb59345d8ba9d2f6610df6441d63082f46021c1685de24716c22f55383

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page