Welcome to yellowbrick’s documentation!

Yellowbrick is a suite of visual analysis and diagnostic tools to facilitate feature selection, model selection, and parameter tuning for machine learning. All visualizations are generated in Matplotlib.

Custom yellowbrick visualization tools include:

Tools for feature analysis and selection

• Boxplots (“box-and-whisker” plots)
• Violinplots
• Histograms
• Scatter plot matrices (“sploms”)
• Radial visualizations (“radviz”)
• Parallel coordinates
• Jointplots
• Rank 1D
• Rank 2D

Tools for model evaluation

Classification

• ROC-AUC curves
• Classification heatmaps
• Class balance charts

Regression

• Prediction error plots
• Residual plots
• Most informative features

Clustering

• Silhouettes
• Density measures

Tools for parameter tuning

• Validation curves
• Gridsearch heatmap

Contents:

• Introduction to Yellowbrick
• Quick Start
  • Installation
  • Using Yellowbrick
    • Feature Visualization
    • Model Visualization
• Examples and Tutorials
  • Methods
    • Anscombe’s Quartet
    • Load Datasets for Example Code
    • Feature Analysis
      • Rank2D
      • RadViz
      • Parallel Coordinates
    • Regressor Evaluation
      • Residuals Plot
      • Prediction Error Plot
    • Classifier Evaluation
      • Classification Report
      • ROCAUC
      • ClassBalance
  • Palettes
  • Model Selection
    • Comparing machine learning models with Scikit-Learn and Yellowbrick
    • About the Data
    • Feature Extraction
    • Modeling and Evaluation
      • Common metrics for evaluating classifiers
      • Preliminary Model Evaluation
    • Visual Model Evaluation
    • Reflection
• API Reference
  • yellowbrick package
    • Submodules
    • yellowbrick.anscombe module
    • yellowbrick.base module
    • yellowbrick.bestfit module
    • yellowbrick.classifier module
    • yellowbrick.exceptions module
    • yellowbrick.pipeline module
    • yellowbrick.regressor module
    • yellowbrick.utils module
    • yellowbrick.version module
    • Module contents
  • yellowbrick.text package
    • Submodules
    • yellowbrick.text.base module
    • yellowbrick.text.tsne module
    • yellowbrick.text.freqdist module
  • yellowbrick.style package
    • Submodules
    • yellowbrick.style.colors module
    • yellowbrick.style.palettes module
    • yellowbrick.style.rcmod module
  • yellowbrick.features package
    • Submodules
    • yellowbrick.features.base module
    • yellowbrick.features.pcoords module
    • yellowbrick.features.radviz module
    • yellowbrick.features.rankd module
• User Testing Instructions
  • Step One: Dataset
  • Step Two: Notebook
  • Step Three: Model with Yellowbrick and Scikit-Learn
  • Step Four: Feedback
  • Step Five: Thanks!
• About
  • About Yellowbrick
    • The Model Selection Triple
  • Contributing
    • Git Conventions
  • Name Origin
  • Changelog
    • Version 0.3.3
    • Version 0.3.2
    • Hotfix 0.3.1
    • Version 0.3
    • Version 0.2
    • Version 0.1

Indices and tables

• Index
• Module Index