Skip to content

A suite of visual analysis and diagnostic tools to facilitate feature selection, model selection, and parameter tuning for machine learning.

License

Notifications You must be signed in to change notification settings

chrisemoulton/yellowbrick

 
 

Repository files navigation

Yellowbrick

Build Status Coverage Status Documentation Status Stories in Ready

A suite of visual analysis and diagnostic tools to facilitate feature selection, model selection, and parameter tuning for machine learning.

Tools for feature analysis and selection

  • boxplots (box-and-whisker plots)
  • violinplots
  • histograms
  • scatter plot matrices (sploms)
  • radial visualizations (radviz)
  • parallel coordinates
  • jointplots
  • diagonal correlation matrix

Tools for model evaluation

Classification

  • ROC curves
  • classification heatmaps

Regression

  • prediction error plots
  • residual plots

Tools for parameter tuning

  • validation curves
  • gridsearch heatmap

About

A suite of visual analysis and diagnostic tools to facilitate feature selection, model selection, and parameter tuning for machine learning.

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 94.9%
  • Makefile 5.1%