yellowbrick 1.4


A suite of visual analysis and diagnostic tools for machine learning.

A suite of visual analysis and diagnostic tools for machine learning.

Stars: 3580, Watchers: 3580, Forks: 519, Open Issues: 92

The DistrictDataLabs/yellowbrick repo was created 5 years ago and was last updated 10 hours ago.
The project is very popular with an impressive 3580 github stars!

How to Install yellowbrick

You can install yellowbrick using pip

pip install yellowbrick

or add it to a project with poetry

poetry add yellowbrick

Package Details

The scikit-yb developers
Apache 2
GitHub Repo


  • Scientific/Engineering/Visualization
  • Software Development
  • Software Development/Libraries/Python Modules
No  yellowbrick  pypi packages just yet.


A list of common yellowbrick errors.

Code Examples

Here are some yellowbrick code examples and snippets.

GitHub Issues

The yellowbrick package has 92 open issues on GitHub

  • Learning Curve Documentation
  • BUG: Corrects legend issues other than R2 in PredictionError
  • Diagnostic Plots for Linear Regression Analysis
  • BUG: Adds missing X and Y axes labels in ClassificationReport
  • Argument to PredictionError is overwritten
  • BUG: Fixes axes limit for PredictionError plot #1193
  • Random input feature dropping curve, model selection visualization [issue #1024]
  • Gap Statistic and Davies-Bouldin Index
  • Model selection curve for Random Input Dropout
  • AttributeError: 'KMeans' object has no attribute 'k'
  • PredictionError plot forces min/max limits on the graph that are unhelpful if you have lots of tiny errors
  • PredictionError hard codes score legend to R2 even when other scoring functions are used

See more issues on GitHub

Related Packages & Articles

autoviz 0.1.39

Automatically Visualize any dataset, any size with a single line of code

dtreeviz 1.3.6

A Python 3 library for sci-kit learn, XGBoost, LightGBM, and Spark decision tree visualization

skorch 0.11.0

scikit-learn compatible neural network library for pytorch

seaborn-image 0.5.0

Attractive, descriptive and effective image visualization with seaborn-like API built on top of matplotlib