Published March 17, 2018
| Version 0.6
Software
Open
Yellowbrick v0.6
Authors/Creators
- 1. District Data Labs
Description
Yellowbrick is an open source, pure Python project that extends the scikit-learn API with visual analysis and diagnostic tools. The Yellowbrick API also wraps matplotlib to create publication-ready figures and interactive data explorations while still allowing developers fine-grain control of figures. For users, Yellowbrick can help evaluate the performance, stability, and predictive value of machine learning models and assist in diagnosing problems throughout the machine learning workflow.
Changes
- New Feature! The
FeatureImportancesVisualizer enables the user to visualize the most informative (relative and absolute) features in their model, plotting a bar graph offeature_importances_orcoef_attributes. - New Feature! The
ExplainedVarianceVisualizer produces a plot of the explained variance resulting from a dimensionality reduction to help identify the best tradeoff between number of dimensions and amount of information retained from the data. - New Feature! The
GridSearchVisualizercreates a color plot showing the best grid search scores across two parameters. - New Feature! The
ClassPredictionErrorVisualizer is a heatmap implementation of the class balance visualizer, which provides a way to quickly understand how successfully your classifier is predicting the correct classes. - New Feature! The
ThresholdVisualizerallows the user to visualize the bounds of precision, recall and queue rate at different thresholds for binary targets after a given number of trials. - New
MultiFeatureVisualizerhelper class to provide base functionality for getting the names of features for use in plot annotation. - Adds font size param to the confusion matrix to adjust its visibility.
- Add quick method to the confusion matrix
- Tests: In this version, we've switched from using nose to pytest. Image comparison tests have been added and the visual tests are updated to matplotlib 2.2.0. Test coverage has also been improved for a number of visualizers, including
JointPlot,AlphaPlot,FreqDist,RadViz,ElbowPlot,SilhouettePlot,ConfusionMatrix,Rank1D, andRank2D. - Documentation updates, including discussion of Image Comparison Tests for contributors.
Bug Fixes:
- Fixes the resolve_colors function. You can now pass in a number of colors and a colormap and get back the correct number of colors.
- Fixes
TSNEVisualizerValue Error when no classes are specified. - Adds the circle back to
RadViz! This visualizer has also been updated to ensure there's a visualization even when there are missing values - Updated
RocAucto correctly check the number of classes - Switch from converting structured arrays to ndarrays using
np.copyinstead ofnp.tolistto avoid NumPy deprecation warning. DataVisualizerupdated to removenp.nanvalues and warn the user that nans are not plotted.ClassificationReportno longer has lines that run through the numbers, is more grid-like
Deprecation Warnings:
ScatterPlotVisualizeris being moved to contrib in 0.7DecisionBoundaryVisualizeris being moved to contrib in 0.7
Files
yellowbrick-0.6.zip
Files
(17.9 MB)
| Name | Size | Download all |
|---|---|---|
|
md5:6cccc6ac24b22db64874d42566672577
|
17.9 MB | Preview Download |
Additional details
Related works
- Is documented by
- http://www.scikit-yb.org/en/stable/ (URL)
- Is supplemented by
- https://github.com/DistrictDataLabs/yellowbrick/releases/tag/v0.6 (URL)