Yellowbrick: Machine Learning Visualization

Yellowbrick is an open-source, pure Python project that wraps matplotlib and extends scikit-learn to create visual analysis and diagnostic tools for machine learning. Yellowbrick’s goal is to assist users in evaluating the performance, stability, and predictive value of machine learning models and assist in diagnosing problems throughout the machine learning workflow. Recently, much of this workflow has been automated through grid search methods, standardized APIs, and GUI-based applications. In practice, however, human intuition and guidance can more effectively hone in on quality models than exhaustive search. By visualizing the model selection process, data scientists can steer towards final, explainable models and avoid pitfalls and traps.

Tech Stack

  • Python

About the Engineer

The Yellowbrick project is managed by a technical steering committee of 9 open source contributors, who also generally act as maintainers. There have been over 85 active contributors to the project.

company logo
The Scikit-Yellowbrick Developers
Scikit-Yellowbrick Development Team

Comment test