Skip to main content

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

Project description

Yellowbrick is a suite of visual analysis and diagnostic tools designed to facilitate machine learning with Scikit-Learn. The package includes visualizations that can help users navigate the feature selection process, build intuition around model selection, diagnose common problems like bias, heteroscedasticity, underfit, and overtraining, and support hyperparameter tuning to steer predictive models toward more successful results.

Some of the available tools include:

  • histograms

  • scatter plot matrices

  • parallel coordinates

  • jointplots

  • ROC curves

  • classification heatmaps

  • residual plots

  • validation curves

  • gridsearch heatmaps

For more, please see the full documentation at: http://yellowbrick.readthedocs.org/en/latest/

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

yellowbrick-0.3.1.tar.gz (3.9 MB view hashes)

Uploaded Source

Built Distribution

yellowbrick-0.3.1-py2.py3-none-any.whl (46.5 kB view hashes)

Uploaded Python 2 Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page