yellowbrick 1.5


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

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

Stars: 3870, Watchers: 3870, Forks: 537, Open Issues: 87

The DistrictDataLabs/yellowbrick repo was created 6 years ago and was last updated 2 days ago.
The project is very popular with an impressive 3870 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 87 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.58

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

dtreeviz 2.0.0

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

skorch 0.12.1

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