Skip to main content

Visualizers

Yellowbrick

Yellowbrick is a suite of visual analysis and diagnostic tools designed to facilitate machine learning with Scikit-Learn. The library implements a new core API object, the “Visualizer” that is an Scikit-Learn estimator: an object that learns from data. Like transformers or models, visualizers learn from data by creating a visual representation of the model selection workflow.

Visualizers allow users to steer the model selection process, building intuition around feature engineering, algorithm selection, and hyperparameter tuning. For example, visualizers can help diagnose common problems surrounding model complexity and bias, heteroscedasticity, underfit and overtraining, or class balance issues. By applying visualizers to the model selection workflow, Yellowbrick allows you to steer predictive models to more successful results, faster.

Please see the full documentation at: http://scikit-yb.org/

Visualizers

Visualizers are estimators (objects that learn from data) whose primary objective is to create visualizations that allow insight into the model selection process. In Scikit-Learn terms, they can be similar to transformers when visualizing the data space or wrap an model estimator similar to how the “ModelCV” (e.g. RidgeCV, LassoCV) methods work. The primary goal of Yellowbrick is to create a sensical API similar to Scikit-Learn. Some of our most popular visualizers include:

Feature Visualization

  • Rank Features: single or pairwise ranking of features to detect relationships

  • Parallel Coordinates: horizontal visualization of instances

  • Radial Visualization: separation of instances around a circular plot

  • PCA Projection: projection of instances based on principal components

  • Scatter and Joint Plots: direct data visualization with feature selection

Classification Visualization

  • Class Balance: see how the distribution of classes affects the model

  • Classification Report: visual representation of precision, recall, and F1

  • ROC/AUC Curves: receiver operator characteristics and area under the curve

  • Confusion Matrices: visual description of class decision making

Regression Visualization

  • Prediction Error Plots: find model breakdowns along the domain of the target

  • Residuals Plot: show the difference in residuals of training and test data

  • Alpha Selection: show how the choice of alpha influences regularization

Clustering Visualization

  • K-Elbow Plot: select k using the elbow method and various metrics

  • Silhouette Plot: select k by visualizing silhouette coefficient values

Text Visualization

  • Term Frequency: visualize the frequency distribution of terms in the corpus

  • TSNE: use stochastic neighbor embedding to project documents.

… and more! Visualizers are being added all the time; be sure to check the examples (or even the develop branch) and feel free to contribute your ideas for new Visualizers!

Release files for yellowbrick 0.5

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for yellowbrick 0.5
File Size Uploaded
yellowbrick-0.5.tar.gz 15.3 MB Details

Built distribution (wheel)

Table of built distributions (wheels) for yellowbrick 0.5
File Interpreter ABI Platform
yellowbrick-0.5-py2.py3-none-any.whl Python 2, Python 3 none any Details

Total release size: 15.4 MB

Release files / yellowbrick-0.5.tar.gz

Download URL yellowbrick-0.5.tar.gz
Size 15.3 MB
Tags Source
SHA-256 checksum
How to use checksums
553187969ebc349f1a9733cd0d86d9f54d974141951411ac774510ba8b72b24a
BLAKE2b-256 checksum
How to use checksums
162c968efb66a0583efc1a35dc92a89b694e4c97f87d144a00d7c72e013628b9
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No

Release files / yellowbrick-0.5-py2.py3-none-any.whl

Download URL yellowbrick-0.5-py2.py3-none-any.whl
Size 151.7 kB
Tags Python 2 Python 3
SHA-256 checksum
How to use checksums
898f5e2b397b5a2d9424ce952f315c4eb44e789b9fde37ee0c5328c37d015104
BLAKE2b-256 checksum
How to use checksums
25d773de84d138fb32ceedbdbe064c9939b5e09a599cc574ea1b32ecc01226f2
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No

Release history Release notifications | RSS feed

1.5

2 release files

1.4

2 release files

1.3

2 release files

1.2.1

2 release files

1.2

3 release files

1.1

2 release files

1.0.1

2 release files

1.0

2 release files

0.9.1

2 release files

0.9

2 release files

0.8

2 release files

0.7

2 release files

0.6

2 release files

This release

0.5 This release

2 release files

0.4.2

2 release files

0.4.1

2 release files

0.4

2 release files

0.3.3

2 release files

0.3.2

2 release files

0.3.1

2 release files

0.1

2 release files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page