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/ particularly the quick start guide
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
Manifold Visualization: high dimensional visualization with manifold learning
Feature Importances: rank features based on their in-model performance
Recursive Feature Elimination: find the best subset of features by importance
Joint Plots: direct data visualization with feature selection
Classification Visualization
Class Balance: see how the distribution of classes affects the model
Class Prediction Error: shows error and support in classification
Classification Report: visual representation of precision, recall, and F1
ROC/AUC Curves: receiver operator characteristics and area under the curve
Precision-Recall Curves: precision vs recall for different probability thresholds
Confusion Matrices: visual description of class decision making
Discrimination Threshold: find a threshold that best separates binary classes
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
Intercluster Distance Maps: show relative distance and size of clusters
Model Selection Visualization
Validation Curve: tune a model with respect to a single hyperparameter
Learning Curve: show if a model might benefit from more data or less complexity
Text Visualization
Term Frequency: visualize the frequency distribution of terms in the corpus
t-SNE Corpus Visualization: use stochastic neighbor embedding to project documents
Dispersion Plot: visualize how key terms are dispersed throughout a corpus
Target Visualization
Feature Correlation: visualize the correlation between the dependent variables and the target
… 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!
3 maintainers
Metadata
3 maintainers
Release files for yellowbrick 0.9
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| yellowbrick-0.9.tar.gz | 30.1 MB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| yellowbrick-0.9-py2.py3-none-any.whl | Python 3, Python 2 | none | any | Details |
Total release size: 30.4 MB
Release files / yellowbrick-0.9.tar.gz
| Download URL | yellowbrick-0.9.tar.gz |
|---|---|
| Size | 30.1 MB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
e4ca98068da958dba27900a315868557c9f6aa759453b25aebbc7e5fda7a61fb
|
|
BLAKE2b-256 checksum How to use checksums |
15abd9d40b35792c19863780abb8e8cfef03f75c643705bd1d4512203f6e01d3
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/1.10.0 pkginfo/1.4.2 requests/2.19.1 setuptools/39.2.0 requests-toolbelt/0.8.0 tqdm/4.19.8 CPython/3.6.2
|
Release files / yellowbrick-0.9-py2.py3-none-any.whl
| Download URL | yellowbrick-0.9-py2.py3-none-any.whl |
|---|---|
| Size | 282.0 kB |
| Tags | Python 2 Python 3 |
|
SHA-256 checksum How to use checksums |
6099d38e35364fa1f14db6d43592d75706c14b36bee12e106cc732f4a5dced1b
|
|
BLAKE2b-256 checksum How to use checksums |
f0b3fffa077157c38e729eb033fdd0ad57fbe92db1aac303c69b283ce10deb9b
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/1.10.0 pkginfo/1.4.2 requests/2.19.1 setuptools/39.2.0 requests-toolbelt/0.8.0 tqdm/4.19.8 CPython/3.6.2
|
