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.3.2.tar.gz (88.0 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.3.2-py3-none-any.whl (102.3 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: dtreeviz-2.3.2.tar.gz
  • Upload date:
  • Size: 88.0 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.12.4

File hashes

Hashes for dtreeviz-2.3.2.tar.gz
Algorithm Hash digest
SHA256 d8413a01373237150fcccb763e469384fbd055a627517d77dbb0190a1403f059
MD5 eb9782b9effc96ab1236d249bfa3498b
BLAKE2b-256 9c0785c7f7e5d25fef6281d7a89ac097271885e7d288628a591ca3ce38a40b03

See more details on using hashes here.

File details

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

File metadata

  • Download URL: dtreeviz-2.3.2-py3-none-any.whl
  • Upload date:
  • Size: 102.3 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.12.4

File hashes

Hashes for dtreeviz-2.3.2-py3-none-any.whl
Algorithm Hash digest
SHA256 df68982726d47fcaa1f0f7306a5a550504697c4681c4245dd5ab2ddba0e270bd
MD5 f21350f0ca69f48df6061f6b128db4c3
BLAKE2b-256 0a593de94d59b3312e39d4b93295262d8f4516e8871864427035b1a9c77be93c

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