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.0.tar.gz (75.9 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

dtreeviz-2.2.0-py3-none-any.whl (90.5 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: dtreeviz-2.2.0.tar.gz
  • Upload date:
  • Size: 75.9 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.0.tar.gz
Algorithm Hash digest
SHA256 0d87b00e78c95b45f24fb6f3a9efbbb34169a8d51aa7b5c7194bf4431443d4c8
MD5 4eed23611ae11cd6629d5b9c71be80d6
BLAKE2b-256 a5daddc3f75762c7342c91f3f1388dabcbd1a852ccaafcabd2994900cf2eb934

See more details on using hashes here.

File details

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

File metadata

  • Download URL: dtreeviz-2.2.0-py3-none-any.whl
  • Upload date:
  • Size: 90.5 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.0-py3-none-any.whl
Algorithm Hash digest
SHA256 8862e25e0124ca6db8d4e6bc1d81570e8cae8fb59ef55ed778c748957981e572
MD5 8ed3895dae18d3f619751b0f8e077685
BLAKE2b-256 6a81c298e05e7969a91995032af6316f5ac77e5aca1f0d78e978a2fe39d11be0

See more details on using hashes here.

Supported by

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