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/

Supported by

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