Skip to content
Go to file
Cannot retrieve contributors at this time
164 lines (114 sloc) 7.35 KB

Yellowbrick: Machine Learning Visualization


Yellowbrick extends the Scikit-Learn API to make model selection and hyperparameter tuning easier. Under the hood, it's using Matplotlib.

Recommended Learning Path

  1. Check out the :doc:`quickstart`, try the :doc:`tutorial`, and check out the :doc:`oneliners`.
  2. Use Yellowbrick in your work, referencing the :doc:`api/index` for assistance with specific visualizers and detailed information on optional parameters and customization options.
  3. Star us on GitHub and follow us on Twitter (@scikit_yb) so that you'll hear about new visualizers as soon as they're added.


Interested in contributing to Yellowbrick? Yellowbrick is a welcoming, inclusive project and we would love to have you. We follow the Python Software Foundation Code of Conduct.

No matter your level of technical skill, you can be helpful. We appreciate bug reports, user testing, feature requests, bug fixes, product enhancements, and documentation improvements.

Check out the :doc:`contributing/index` guide!

If you've signed up to do user testing, head over to the :doc:`evaluation`.

Please consider joining the Google Groups Listserv listserve so you can respond to questions.

Thank you for your contributions!

Concepts & API


The primary goal of Yellowbrick is to create a sensical API similar to Scikit-Learn.

Visualizers are the core objects in Yellowbrick. They are similar to transformers in Scikit-Learn. Visualizers can wrap a model estimator - similar to how the "ModelCV" (e.g. RidgeCV, LassoCV) methods work.

Some of our most popular visualizers include:

Feature Visualization

Classification Visualization

Regression Visualization

Clustering Visualization

Model Selection Visualization

Target Visualization

Text Visualization

... and more! Visualizers are being added all the time. Check the examples (or even the develop branch). Feel free to contribute your ideas for new Visualizers!

Getting Help

Can't get someting to work? Here are places you can find help.

  1. The docs (you're here!).
  2. Stack Overflow. If you ask a question, please tag it with "yellowbrick".
  3. The Yellowbrick Google Groups Listserv.
  4. You can also Tweet or direct message us on Twitter @scikit_yb.

Find a Bug?

Check if there's already an open issue on the topic. If needed, file an issue.

Open Source

The Yellowbrick license is an open source Apache 2.0 license. Yellowbrick enjoys a very active developer community; please consider :doc:`contributing/index`!

Yellowbrick is hosted on GitHub. The issues and pull requests are tracked there.

Table of Contents

.. toctree::
   :maxdepth: 2


Indices and tables

You can’t perform that action at this time.