A suite of visual analysis and diagnostic tools for machine learning.
.. -- mode: rst --
|Visualizers|_
.. |Visualizers| image:: http://www.scikit-yb.org/en/latest/_images/visualizers.png :width: 800 px .. _Visualizers: http://www.scikit-yb.org/
Yellowbrick is a suite of visual analysis and diagnostic tools designed to facilitate machine learning with Scikit-Learn. The library implements a new core API object, the "Visualizer" that is an Scikit-Learn estimator: an object that learns from data. Like transformers or models, visualizers learn from data by creating a visual representation of the model selection workflow.
Visualizers allow users to steer the model selection process, building intuition around feature engineering, algorithm selection, and hyperparameter tuning. For example, visualizers can help diagnose common problems surrounding model complexity and bias, heteroscedasticity, underfit and overtraining, or class balance issues. By applying visualizers to the model selection workflow, Yellowbrick allows you to steer predictive models to more successful results, faster.
Please see the full documentation at: http://scikit-yb.org/ particularly the quick start guide <http://www.scikit-yb.org/en/latest/quickstart.html>
_
Visualizers are estimators (objects that learn from data) whose primary objective is to create visualizations that allow insight into the model selection process. In Scikit-Learn terms, they can be similar to transformers when visualizing the data space or wrap an model estimator similar to how the “ModelCV” (e.g. RidgeCV, LassoCV) methods work. The primary goal of Yellowbrick is to create a sensical API similar to Scikit-Learn. Some of our most popular visualizers include:
.. RidgeCV: http://scikit-learn.org/stable/modules/generated/sklearn.linearmodel.RidgeCV.html .. LassoCV: http://scikit-learn.org/stable/modules/generated/sklearn.linearmodel.LassoCV.html
Feature Visualization ~~~~~~~~~~~~~~~~~~~~~
Classification Visualization ~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Regression Visualization ~~~~~~~~~~~~~~~~~~~~~~~~
Clustering Visualization ~~~~~~~~~~~~~~~~~~~~~~~~
Model Selection Visualization ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Text Visualization ~~~~~~~~~~~~~~~~~~
Target Visualization ~~~~~~~~~~~~~~~~~~~~
... and more! Visualizers are being added all the time; be sure to check the examples_ (or even the develop_ branch) and feel free to contribute your ideas for new Visualizers!
.. _examples: http://www.scikit-yb.org/en/latest/api/index.html .. _develop: https://github.com/districtdatalabs/yellowbrick/tree/develop