Version 1.5

  • Tag: v1.5

  • Deployed Sunday, August 21, 2022

  • Current Contributors: Stefanie Molin, Prema Roman, Sangam Swadik, David Gilbertson, Larry Gray, Benjamin Bengfort, @admo1, @charlesincharge, Uri Nussbaum, Patrick Deziel, Rebecca Bilbro

  • Added WordCorrelationPlot Visualizer

  • Built tests for using sklearn pipeline with visualizers

  • Allowed Marker Style to be specified in Validation Curve Visualizer

  • Fixed get_params for estimator wrapper to prevent AttributeError

  • Updated missing values visualizer to handle multiple data types and work on both numpy arrays and pandas data frames.

  • Added pairwise distance metrics to scoring metrics in KElbowVisualizer

  • Pegged Numba to v0.55.2

  • Updated Umap to v0.5.3

  • Fixed Missing labels in classification report visualizer

  • Updated Numpy to v1.22.0

  • The Spanish language Yellowbrick docs are now live:

  • Added Dropping curve documentation

  • Added new example Notebook for Regression Visualizers

  • Fixed Typo in PR section of getting started docs

  • Fixed Typo in rank docs

  • Updated docstring in utility file

  • Clarified how to run ‘make html’ in PR template

  • Added ability to run linting Actions on PRs

  • Implemented black code formatting as pre-commit hook

Version 1.4

  • Tag: v1.4

  • Deployed Saturday, February 19, 2022

  • Current Contributors: Benjamin Bengfort, Larry Gray, Rebecca Bilbro, @pkaf, Antonio Carlos Falcão Petri, Aarni Koskela, Prema Roman, Nathan Danielsen, Eleni Markou, Patrick Deziel, Adam Morris, Hung-Tien Huang, @charlesincharge

  • Upgrade dependencies to support sklearn v1.0, Numpy 1.20+, Scipy 1.6, nltk 3.6.7, and Matplotlib 3.4.1

  • Implement new set_params and get_params on ModelVisualizers to ensure wrapped estimator is being correctly accessed via the new Estimator methods.

  • Fix the test dependencies to prevent variability in CI (must periodically review dependencies to ensure we’re testing what our users are experiencing).

  • Change model param to estimator param to ensure that Visualizer arguments match their property names so that inspect works with get and set params and other scikit-learn utility functions.

  • Improved argmax handling in DiscriminationThreshold Visualizer

  • Improved error handling in FeatureImportances Visualizer

  • Gave option to remove colorer from ClassificationReport Visualizer

  • Allowed for more flexible KElbow colors that use default palette by default

  • Import scikit-learn private API _safe_indexing without error.

  • Remove any calls to set_params in Visualizer __init__ methods.

  • Modify test fixtures and baseline images to accommodate new sklearn implementation

  • Temporarily set the numpy dependency to be less than 1.20 because this is causing Pickle issues with joblib and umap

  • Add shuffle=True argument to any CV class that uses a random seed.

  • Set our CI matrix to Python and Miniconda 3.7 and 3.8

  • Fixed score label display in PredictionError Visualizer

  • Fixed axes limit in PredictionError Visualizer

  • Fixed KElbowVisualizer to handle null cluster encounters

  • Fixed broken url to pytest fixtures

  • Fixed random_state to be in sync with PCA transformer

  • Fixed the inability to place FeatureCorrelations into subplots

  • Fixed hanging printing impacting model visualizers

  • Fixed error handling when decision function models encounter binary data

  • Fixed missing code in

  • Updated status badges for build result and code coverage

  • Removed deprecated pytest-runner from testing

  • Replaced Travis with Github Actions

  • Changed our master branch to the main branch

  • Created a release issue template

  • Updated our CI to test Python 3.8 and 3.9

  • Managed test warnings

  • Adds .gitattributes to fix handle white space changes

  • Updated to use add_css_file for documentation because of deprecation of add_stylesheet

  • Added a Sphinx build to GitHub Actions for ensuring that the docs build correctly

  • Switched to a YB-specific data lake for datasets storage

Version 1.3.post1

  • Tag: v1.3.post1

  • Deployed Saturday, February 13, 2021

  • Current Contributors: Rebecca Bilbro, Benjamin Bengfort, (EJ) Vivek Pandey

Fixes hanging print impacting ModelVisualizers.

Version 1.3

  • Tag: v1.3

  • Deployed Tuesday, February 9, 2021

  • Current Contributors: Benjamin Bengfort, Rebecca Bilbro, Paul Johnson, Philippe Billet, Prema Roman, Patrick Deziel

This version primarily repairs the dependency issues we faced with scipy 1.6, scikit-learn 0.24 and Python 3.6 (or earlier). As part of the rapidly changing Python library landscape, we’ve been forced to react quickly to dependency changes, even where those libraries have been responsibly issuing future and deprecation warnings.

Major Changes:
  • Implement new set_params and get_params on ModelVisualizers to ensure wrapped estimator is being correctly accessed via the new Estimator methods.

  • Freeze the test dependencies to prevent variability in CI (must periodically review dependencies to ensure we’re testing what our users are experiencing).

  • Change model param to estimator param to ensure that Visualizer arguments match their property names so that inspect works with get and set params and other scikit-learn utility functions.

Minor Changes:
  • Import scikit-learn private API _safe_indexing without error.

  • Remove any calls to set_params in Visualizer __init__ methods.

  • Modify test fixtures and baseline images to accommodate new sklearn implementation

  • Set the numpy dependency to be less than 1.20 because this is causing Pickle issues with joblib and umap

  • Add shuffle=True argument to any CV class that uses a random seed.

  • Set our CI matrix to Python and Miniconda 3.7 and 3.8

  • Correction in README regarding ModelVisualizer API.

Hotfix 1.2.1

  • Tag: v1.2.1

  • Deployed Friday, January 15, 2020

  • Contributors: Rebecca Bilbro, Benjamin Bengfort, Paul Johnson, Matt Harrison

On December 22, 2020, scikit-learn released version 0.24 which deprecated the external use of scikit-learn’s internal utilities such as safe_indexing. Unfortunately, Yellowbrick depends on a few of these utilities and must refactor our internal code base to port this functionality or work around it. To ensure that Yellowbrick continues to work when installed via pip, we have temporarily changed our scikit-learn dependency to be less than 0.24. We will update our dependencies on the v1.3 release when we have made the associated fixes.

Version 1.2

  • Tag: v1.2

  • Deployed Friday, October 9, 2020

  • Current Contributors: Rebecca Bilbro, Larry Gray, Vladislav Skripniuk, David Landsman, Prema Roman, @aldermartinez, Tan Tran, Benjamin Bengfort, Kellen Donohue, Kristen McIntyre, Tony Ojeda, Edwin Schmierer, Adam Morris, Nathan Danielsen

Major Changes:
  • Added Q-Q plot as side-by-side option to the ResidualsPlot visualizer.

  • More robust handling of binary classification in ROCAUC visualization, standardizing the way that classifiers with predict_proba and decision_function methods are handling. A binary hyperparameter was added to the visualizer to ensure correct interpretation of binary ROCAUC plots.

  • Fixes to ManualAlphaSelection to move it from prototype to prime time including documentation, tests, and quick method. This method allows users to perform alpha selection visualization on non-CV estimators.

  • Removal of AppVeyor from the CI matrix after too many out-of-core (non-Yellowbrick) failures with setup and installation on the VisualStudio images. Yellowbrick CI currently omits Windows and Miniconda from the test matrix and we are actively looking for new solutions.

  • Third party estimator wrapper in contrib to provide enhanced support for non-scikit-learn estimators such as those in Keras, CatBoost, and cuML.

Minor Changes:
  • Allow users to specify colors for the PrecisionRecallCurve.

  • Update ClassificationScoreVisualizer base class to have a class_colors_ learned attribute instead of a colors property; additional polishing of multi-class colors in PrecisionRecallCurve, ROCAUC, and ClassPredictionError.

  • Update KElbowVisualizer fit method and quick method to allow passing sample_weight parameter through the visualizer.

  • Enhancements to classification documentation to better discuss precision and recall and to diagnose with PrecisionRecallCurve and ClassificationReport visualizers.

  • Improvements to CooksDistance visualizer documentation.

  • Corrected KElbowVisualizer label and legend formatting.

  • Typo fixes to ROCAUC documentation, labels, and legend. Typo fix to Manifold documentation.

  • Use of tight_layout accessing the Visualizer figure property to finalize images and resolve discrepancies in plot directive images in documentation.

  • Add get_param_names helper function to identify keyword-only parameters that belong to a specific method.

  • Splits package namespace for yellowbrick.regressor.residuals to move PredictionError to its own module, yellowbrick.regressor.prediction_error.

  • Update tests to use SVC instead of LinearSVC and correct KMeans scores based on updates to scikit-learn v0.23.

  • Continued maintenance and management of baseline images following dependency updates; removal of mpl.cbook dependency.

  • Explicitly include license file in source distribution via

  • Fixes to some deprecation warnings from sklearn.metrics.

  • Testing requirements depends on Pandas v1.0.4 or later.

  • Reintegrates pytest-spec and verbose test logging, updates pytest dependency to v0.5.0 or later.

  • Added Pandas v0.20 or later to documentation dependencies.

Version 1.1

  • Tag: v1.1

  • Deployed Wednesday, February 12, 2020

  • Contributors: Benjamin Bengfort, Rebecca Bilbro, Kristen McIntyre, Larry Gray, Prema Roman, Adam Morris, Shivendra Sharma, Michael Chestnut, Michael Garod, Naresh Bachwani, Piyush Gautam, Daniel Navarrete, Molly Morrison, Emma Kwiecinska, Sarthak Jain, Tony Ojeda, Edwin Schmierer, Nathan Danielsen

Major Changes:
  • Quick methods (aka Oneliners), which return a fully fitted finalized visualizer object in only a single line, are now implemented for all Yellowbrick Visualizers. Test coverage has been added for all quick methods. The documentation has been updated to document and demonstrate the usage of the quick methods.

  • Added Part of Speech tagging for raw text using spaCy and NLTK to POSTagVisualizer.

Minor Changes:
  • Adds Board of Directors minutes for Spring meeting.

  • Miscellaneous documentation corrections and fixes.

  • Miscellaneous CI and testing corrections and fixes.

Hotfix 1.0.1

  • Tag: v1.0.1

  • Deployed Sunday, October 6, 2019

  • Contributors: Benjamin Bengfort, Rebecca Bilbro, Kristen McIntyre


Major API change: the poof() method is now deprecated, please use show() instead. After a significant discussion with community members we have deprecated our original “make the magic happen” method due to concerns about the usage of the word. We’ve renamed the original method to and created a stub method with the original name that issues a deprecation warning and calls show().

  • Changes poof() to show().

  • Updated clustering and regression example notebooks.

  • Fixes a syntax error in Python 3.5 and earlier.

  • Updated Manifold documentation to fix example bug.

  • Added advisors names to the release changelog.

  • Adds advisory board minutes for Fall 2019.

  • Updates our Travis-CI semi-secure token for Slack integration.

Version 1.0

  • Tag: v1.0

  • Deployed Wednesday, August 28, 2019

  • Contributors: Benjamin Bengfort, Rebecca Bilbro, Nathan Danielsen, Kristen McIntyre, Larry Gray, Prema Roman, Adam Morris, Tony Ojeda, Edwin Schmierer, Carl Dawson, Daniel Navarrete, Francois Dion, Halee Mason, Jeff Hale, Jiayi Zhang, Jimmy Shah, John Healy, Justin Ormont, Kevin Arvai, Michael Garod, Mike Curry, Nabanita Dash, Naresh Bachwani, Nicholas A. Brown, Piyush Gautam, Pradeep Singh, Rohit Ganapathy, Ry Whittington, Sangarshanan, Sourav Singh, Thomas J Fan, Zijie (ZJ) Poh, Zonghan, Xie


Python 2 Deprecation: Please note that this release deprecates Yellowbrick’s support for Python 2.7. After careful consideration and following the lead of our primary dependencies (NumPy, scikit-learn, and Matplolib), we have chosen to move forward with the community and support Python 3.4 and later.

Major Changes:
  • New JointPlot visualizer that is specifically designed for machine learning. The new visualizer can compare a feature to a target, features to features, and even feature to feature to target using color. The visualizer gives correlation information at a glance and is designed to work on ML datasets.

  • New PosTagVisualizer is specifically designed for diagnostics around natural language processing and grammar-based feature extraction for machine learning. This new visualizer shows counts of different parts-of-speech throughout a tagged corpus.

  • New datasets module that provide greater support for interacting with Yellowbrick example datasets including support for Pandas, npz, and text corpora.

  • Management repository for Yellowbrick example data, yellowbrick-datasets.

  • Add support for matplotlib 3.0.1 or greater.

  • UMAPVisualizer as an alternative manifold to TSNE for corpus visualization that is fast enough to not require preprocessing PCA or SVD decomposition and preserves higher order similarities and distances.

  • Added ..plot:: directives to the documentation to automatically build the images along with the docs and keep them as up to date as possible. The directives also include the source code making it much simpler to recreate examples.

  • Added target_color_type functionality to determine continuous or discrete color representations based on the type of the target variable.

  • Added alpha param for both test and train residual points in ResidualsPlot.

  • Added frameon param to Manifold.

  • Added frequency sort feature to PosTagVisualizer.

  • Added elbow detection using the “kneedle” method to the KElbowVisualizer.

  • Added governance document outlining new Yellowbrick structure.

  • Added CooksDistance regression visualizer.

  • Updated DataVisualizer to handle target type identification.

  • Extended DataVisualizer and updated its subclasses.

  • Added ProjectionVisualizer base class.

  • Restructured, yellowbrick.features, and yellowbrick.model_selection API.

  • Restructured regressor and classifier API.

Minor Changes:
  • Updated Rank2D to include Kendall-Tau metric.

  • Added user specification of ISO F1 values to PrecisionRecallCurve and updated the quick method to accept train and test splits.

  • Added code review checklist and conventions to the documentation and expanded the contributing docs to include other tricks and tips.

  • Added polish to missing value visualizers code, tests, and documentation.

  • Improved RankD tests for better coverage.

  • Added quick method test for DispersionPlot visualizer.

  • BugFix: fixed resolve colors bug in TSNE and UMAP text visualizers and added regression tests to prevent future errors.

  • BugFix: Added support for Yellowbrick palettes to return colormap.

  • BugFix: fixed PrecisionRecallCurve visual display problem with multi-class labels.

  • BugFix: fixed the RFECV step display bug.

  • BugFix: fixed error in distortion score calculation.

  • Extended FeatureImportances documentation and tests for stacked importances and added a warning when stack should be true.

  • Improved the documentation readability and structure.

  • Refreshed the and added testing and documentation READMEs.

  • Updated the gallery to generate thumbnail-quality images.

  • Updated the example notebooks and created a quickstart notebook.

  • Fixed broken links in the documentation.

  • Enhanced the SilhouetteVisualizer with legend and color parameter, while also move labels to the y-axis.

  • Extended FeatureImportances docs/tests for stacked importances.

  • Documented the script.

  • Added JOSS citation for “Yellowbrick: Visualizing the Scikit-Learn Model Selection Process”.

  • Added new pull request (PR) template.

  • Added alpha param to PCA Decomposition Visualizer.

  • Updated documentation with affiliations.

  • Added a windows_tol for the visual unittest suite.

  • Added stacked barchart to PosTagVisualizer.

  • Let users set colors for FreqDistVisualizer and other ax_bar visualizers.

  • Updated Manifold to extend ProjectionVisualizer.

  • Check if an estimator is already fitted before calling fit method.

  • Ensure poof returns ax.

Compatibility Notes:
  • This version provides support for matplotlib 3.0.1 or greater and drops support for matplotlib versions less than 2.0.

  • This version drops support for Python 2

Hotfix 0.9.1

This hotfix adds matplotlib3 support by requiring any version of matplotlib except for 3.0.0 which had a backend bug that affected Yellowbrick.

  • Tag: v0.9.1

  • Deployed: Tuesday, February 5, 2019

  • Contributors: Benjamin Bengfort, Rebecca Bilbro, Ian Ozsvald, Francois Dion

Version 0.9

  • Tag: v0.9

  • Deployed: Wednesday, November 14, 2018

  • Contributors: Rebecca Bilbro, Benjamin Bengfort, Zijie (ZJ) Poh, Kristen McIntyre, Nathan Danielsen, David Waterman, Larry Gray, Prema Roman, Juan Kehoe, Alyssa Batula, Peter Espinosa, Joanne Lin, @rlshuhart, @archaeocharlie, @dschoenleber, Tim Black, @iguk1987, Mohammed Fadhil, Jonathan Lacanlale, Andrew Godbehere, Sivasurya Santhanam, Gopal Krishna

Major Changes:
  • Target module added for visualizing dependent variable in supervised models.

  • Prototype missing values visualizer in contrib module.

  • BalancedBinningReference visualizer for thresholding unbalanced data (undocumented).

  • CVScores visualizer to instrument cross-validation.

  • FeatureCorrelation visualizer to compare relationship between a single independent variable and the target.

  • ICDM visualizer, intercluster distance mapping using projections similar to those used in pyLDAVis.

  • PrecisionRecallCurve visualizer showing the relationship of precision and recall in a threshold-based classifier.

  • Enhanced FeatureImportance for multi-target and multi-coefficient models (e.g probabilistic models) and allows stacked bar chart.

  • Adds option to plot PDF to ResidualsPlot histogram.

  • Adds document boundaries option to DispersionPlot and uses colored markers to depict class.

  • Added alpha parameter for opacity to the scatter plot visualizer.

  • Modify KElbowVisualizer to accept a list of k values.

  • ROCAUC bugfix to allow binary classifiers that only have a decision function.

  • TSNE bugfix so that title and size params are respected.

  • ConfusionMatrix bugfix to correct percentage displays adding to 100.

  • ResidualsPlot bugfix to ensure specified colors are both in histogram and scatterplot.

  • Fixed unicode decode error on Py2 compatible Windows using Hobbies corpus.

  • Require matplotlib 1.5.1 or matplotlib 2.0 (matplotlib 3.0 not supported yet).

  • Deprecated percent and sample_weight arguments to ConfusionMatrix fit method.

  • Yellowbrick now depends on SciPy 1.0 and scikit-learn 0.20.

Minor Changes:
  • Removed hardcoding of SilhouetteVisualizer axes dimensions.

  • Audit classifiers to ensure they conform to score API.

  • Fix for Manifold fit_transform bug.

  • Fixed Manifold import bug.

  • Started reworking datasets API for easier loading of examples.

  • Added Timer utility for keeping track of fit times.

  • Added slides to documentation for teachers teaching ML/Yellowbrick.

  • Added an FAQ to the documentation.

  • Manual legend drawing utility.

  • New examples notebooks for regression and clustering.

  • Example of interactive classification visualization using ipywidgets.

  • Example of using Yellowbrick with PyTorch.

  • Repairs to ROCAUC tests and binary/multiclass ROCAUC construction.

  • Rename tests/ to tests/ to prevent NumPy errors.

  • Improves ROCAUC, KElbowVisualizer, and SilhouetteVisualizer documentation.

  • Fixed visual display bug in JointPlotVisualizer.

  • Fixed image in JointPlotVisualizer documentation.

  • Clear figure option to poof.

  • Fix color plotting error in residuals plot quick method.

  • Fixed bugs in KElbowVisualizer, FeatureImportance, Index, and Datasets documentation.

  • Use LGTM for code quality analysis (replacing Landscape).

  • Updated contributing docs for better PR workflow.

  • Submitted JOSS paper.

Version 0.8

  • Tag: v0.8

  • Deployed: Thursday, July 12, 2018

  • Contributors: Rebecca Bilbro, Benjamin Bengfort, Nathan Danielsen, Larry Gray, Prema Roman, Adam Morris, Kristen McIntyre, Raul Peralta, Sayali Sonawane, Alyssa Riley, Petr Mitev, Chris Stehlik, @thekylesaurus, Luis Carlos Mejia Garcia, Raul Samayoa, Carlo Mazzaferro

Major Changes:
  • Added Support to ClassificationReport - @ariley1472

  • We have an updated Image Gallery - @ralle123

  • Improved performance of ParallelCoordinates Visualizer @ thekylesaurus

  • Added Alpha Transparency to RadViz Visualizer @lumega

  • CVScores Visualizer - @pdamodaran

  • Added fast and alpha parameters to ParallelCoordinates visualizer @bbengfort

  • Make support an optional parameter for ClassificationReport @lwgray

  • Bug Fix for Usage of multidimensional arrays in FeatureImportance visualizer @rebeccabilbro

  • Deprecate ScatterVisualizer to contrib @bbengfort

  • Implements histogram alongside ResidualsPlot @bbengfort

  • Adds biplot to the PCADecomposition visualizer @RaulPL

  • Adds Datasaurus Dataset to show importance of visualizing data @lwgray

  • Add DispersionPlot Plot @lwgray

Minor Changes:
  • Fix grammar in tutorial.rst - @chrisfs

  • Added Note to tutorial indicating subtle differences when working in Jupyter notebook - @chrisfs

  • Update Issue template @bbengfort

  • Added Test to check for NLTK postag data availability - @Sayali

  • Clarify quick start documentation @mitevpi

  • Deprecated DecisionBoundary

  • Threshold Visualization aliases deprecated

Version 0.7

  • Tag: v0.7

  • Deployed: Thursday, May 17, 2018

  • Contributors: Benjamin Bengfort, Nathan Danielsen, Rebecca Bilbro, Larry Gray, Ian Ozsvald, Jeremy Tuloup, Abhishek Bharani, Raúl Peralta Lozada, Tabishsada, Kristen McIntyre, Neal Humphrey


  • New Feature! Manifold visualizers implement high-dimensional visualization for non-linear structural feature analysis.

  • New Feature! There is now a model_selection module with LearningCurve and ValidationCurve visualizers.

  • New Feature! The RFECV (recursive feature elimination) visualizer with cross-validation visualizes how removing the least performing features improves the overall model.

  • New Feature! The VisualizerGrid is an implementation of the MultipleVisualizer that creates axes for each visualizer using plt.subplots, laying the visualizers out as a grid.

  • New Feature! Added yellowbrick.datasets to load example datasets.

  • New Experimental Feature! An experimental StatsModelsWrapper was added to yellowbrick.contrib.statsmodels that will allow user to use StatsModels estimators with visualizers.

  • Enhancement! ClassificationReport documentation to include more details about how to interpret each of the metrics and compare the reports against each other.

  • Enhancement! Modifies scoring mechanism for regressor visualizers to include the R2 value in the plot itself with the legend.

  • Enhancement! Updated and renamed the ThreshViz to be defined as DiscriminationThreshold, implements a few more discrimination features such as F1 score, maximizing arguments and annotations.

  • Enhancement! Update clustering visualizers and corresponding distortion_score to handle sparse matrices.

  • Added code of conduct to meet the GitHub community guidelines as part of our contributing documentation.

  • Added is_probabilistic type checker and converted the type checking tests to pytest.

  • Added a contrib module and DecisionBoundaries visualizer has been moved to it until further work is completed.

  • Numerous fixes and improvements to documentation and tests. Add academic citation example and Zenodo DOI to the Readme.

Bug Fixes:
  • Adds RandomVisualizer for testing and add it to the VisualizerGrid test cases.

  • Fix / update tests in to remove hardcoded data.

Deprecation Warnings:
  • ScatterPlotVisualizer is being moved to contrib in 0.8

  • DecisionBoundaryVisualizer is being moved to contrib in 0.8

  • ThreshViz is renamed to DiscriminationThreshold.

NOTE: These deprecation warnings originally mentioned deprecation in 0.7, but their life was extended by an additional version.

Version 0.6

  • Tag: v0.6

  • Deployed: Saturday, March 17, 2018

  • Contributors: Benjamin Bengfort, Nathan Danielsen, Rebecca Bilbro, Larry Gray, Kristen McIntyre, George Richardson, Taylor Miller, Gary Mayfield, Phillip Schafer, Jason Keung

  • New Feature! The FeatureImportances Visualizer enables the user to visualize the most informative (relative and absolute) features in their model, plotting a bar graph of feature_importances_ or coef_ attributes.

  • New Feature! The ExplainedVariance Visualizer 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 GridSearchVisualizer creates a color plot showing the best grid search scores across two parameters.

  • New Feature! The ClassPredictionError Visualizer 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 ThresholdVisualizer allows 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 MultiFeatureVisualizer helper 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 for 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, and Rank2D.

  • 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 TSNEVisualizer Value 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 RocAuc to correctly check the number of classes

  • Switch from converting structured arrays to ndarrays using np.copy instead of np.tolist to avoid NumPy deprecation warning.

  • DataVisualizer updated to remove np.nan values and warn the user that nans are not plotted.

  • ClassificationReport no longer has lines that run through the numbers, is more grid-like

Deprecation Warnings:
  • ScatterPlotVisualizer is being moved to contrib in 0.7

  • DecisionBoundaryVisualizer is being moved to contrib in 0.7

Version 0.5

  • Tag: v0.5

  • Deployed: Wednesday, August 9, 2017

  • Contributors: Benjamin Bengfort, Rebecca Bilbro, Nathan Danielsen, Carlo Morales, Jim Stearns, Phillip Schafer, Jason Keung

  • Added VisualTestCase.

  • New PCADecomposition Visualizer, which decomposes high-dimensional data into two or three dimensions so that each instance can be plotted in a scatter plot.

  • New and improved ROCAUC Visualizer, which now supports multiclass classification.

  • Prototype DecisionBoundary Visualizer, which is a bivariate data visualization algorithm that plots the decision boundaries of each class.

  • Added Rank1D Visualizer, which is a one-dimensional ranking of features that utilizes the Shapiro-Wilks ranking by taking into account only a single feature at a time (e.g. histogram analysis).

  • Improved PredictionErrorPlot with identity line, shared limits, and R-squared.

  • Updated FreqDist Visualizer to make word features a hyperparameter.

  • Added normalization and scaling to ParallelCoordinates.

  • Added Learning Curve Visualizer, which displays a learning curve based on the number of samples versus the training and cross validation scores to show how a model learns and improves with experience.

  • Added data downloader module to the Yellowbrick library.

  • Complete overhaul of the Yellowbrick documentation; categories of methods are located in separate pages to make it easier to read and contribute to the documentation.

  • Added a new color palette inspired by ANN-generated colors

Bug Fixes:
  • Repairs to PCA, RadViz, FreqDist unit tests

  • Repair to matplotlib version check in JointPlotVisualizer

Hotfix 0.4.2

Update to the deployment docs and package on both Anaconda and PyPI.

  • Tag: v0.4.2

  • Deployed: Monday, May 22, 2017

  • Contributors: Benjamin Bengfort, Jason Keung

Version 0.4.1

This release is an intermediate version bump in anticipation of the PyCon 2017 sprints.

The primary goals of this version were to (1) update the Yellowbrick dependencies (2) enhance the Yellowbrick documentation to help orient new users and contributors, and (3) make several small additions and upgrades (e.g. pulling the Yellowbrick utils into a standalone module).

We have updated the scikit-learn and SciPy dependencies from version 0.17.1 or later to 0.18 or later. This primarily entails moving from from sklearn.cross_validation import train_test_split to from sklearn.model_selection import train_test_split.

The updates to the documentation include new Quickstart and Installation guides, as well as updates to the Contributors documentation, which is modeled on the scikit-learn contributing documentation.

This version also included upgrades to the KMeans visualizer, which now supports not only silhouette_score but also distortion_score and calinski_harabaz_score. The distortion_score computes the mean distortion of all samples as the sum of the squared distances between each observation and its closest centroid. This is the metric that KMeans attempts to minimize as it is fitting the model. The calinski_harabaz_score is defined as ratio between the within-cluster dispersion and the between-cluster dispersion.

Finally, this release includes a prototype of the VisualPipeline, which extends scikit-learn’s Pipeline class, allowing multiple Visualizers to be chained or sequenced together.

  • Tag: v0.4.1

  • Deployed: Monday, May 22, 2017

  • Contributors: Benjamin Bengfort, Rebecca Bilbro, Nathan Danielsen

  • Score and model visualizers now wrap estimators as proxies so that all methods on the estimator can be directly accessed from the visualizer

  • Updated scikit-learn dependency from >=0.17.1 to >=0.18

  • Replaced sklearn.cross_validation with model_selection

  • Updated SciPy dependency from >=0.17.1 to >=0.18

  • ScoreVisualizer now subclasses ModelVisualizer; towards allowing both fitted and unfitted models passed to Visualizers

  • Added CI tests for Python 3.6 compatibility

  • Added new quickstart guide and install instructions

  • Updates to the contributors documentation

  • Added distortion_score and calinski_harabaz_score computations and visualizations to KMeans visualizer.

  • Replaced the property on all of the individual draw methods with a new property on the Visualizer class that ensures all visualizers automatically have axes.

  • Refactored the utils module into a package

  • Continuing to update the docstrings to conform to Sphinx

  • Added a prototype visual pipeline class that extends the scikit-learn pipeline class to ensure that visualizers get called correctly.

Bug Fixes:
  • Fixed title bug in Rank2D FeatureVisualizer

Version 0.4

This release is the culmination of the Spring 2017 DDL Research Labs that focused on developing Yellowbrick as a community effort guided by a sprint/agile workflow. We added several more visualizers, did a lot of user testing and bug fixes, updated the documentation, and generally discovered how best to make Yellowbrick a friendly project to contribute to.

Notable in this release is the inclusion of two new feature visualizers that use few, simple dimensions to visualize features against the target. The JointPlotVisualizer graphs a scatter plot of two dimensions in the data set and plots a best fit line across it. The ScatterVisualizer also uses two features, but also colors the graph by the target variable, adding a third dimension to the visualization.

This release also adds support for clustering visualizations, namely the elbow method for selecting K, KElbowVisualizer and a visualization of cluster size and density using the SilhouetteVisualizer. The release also adds support for regularization analysis using the AlphaSelection visualizer. Both the text and classification modules were also improved with the inclusion of the PosTagVisualizer and the ConfusionMatrix visualizer respectively.

This release also added an Anaconda repository and distribution so that users can conda install yellowbrick. Even more notable, we got Yellowbrick stickers! We’ve also updated the documentation to make it more friendly and a bit more visual; fixing the API rendering errors. All-in-all, this was a big release with a lot of contributions and we thank everyone that participated in the lab!

  • Tag: v0.4

  • Deployed: Thursday, May 4, 2017

  • Contributors: Benjamin Bengfort, Rebecca Bilbro, Nathan Danielsen, Matt Andersen, Prema Roman, Neal Humphrey, Jason Keung, Bala Venkatesan, Paul Witt, Morgan Mendis, Tuuli Morril

  • Part of speech tags visualizer – PosTagVisualizer.

  • Alpha selection visualizer for regularized regression – AlphaSelection

  • Confusion Matrix Visualizer – ConfusionMatrix

  • Elbow method for selecting K vis – KElbowVisualizer

  • Silhouette score cluster visualization – SilhouetteVisualizer

  • Joint plot visualizer with best fit – JointPlotVisualizer

  • Scatter visualization of features – ScatterVisualizer

  • Added three more example datasets: mushroom, game, and bike share

  • Contributor’s documentation and style guide

  • Maintainers listing and contacts

  • Light/Dark background color selection utility

  • Structured array detection utility

  • Updated classification report to use colormesh

  • Added anacondas packaging and distribution

  • Refactoring of the regression, cluster, and classification modules

  • Image based testing methodology

  • Docstrings updated to a uniform style and rendering

  • Submission of several more user studies

Version 0.3.3

Intermediate sprint to demonstrate prototype implementations of text visualizers for NLP models. Primary contributions were the FreqDistVisualizer and the TSNEVisualizer.

The TSNEVisualizer displays a projection of a vectorized corpus in two dimensions using TSNE, a nonlinear dimensionality reduction method that is particularly well suited to embedding in two or three dimensions for visualization as a scatter plot. TSNE is widely used in text analysis to show clusters or groups of documents or utterances and their relative proximities.

The FreqDistVisualizer implements frequency distribution plot that tells us the frequency of each vocabulary item in the text. In general, it could count any kind of observable event. It is a distribution because it tells us how the total number of word tokens in the text are distributed across the vocabulary items.

  • Tag: v0.3.3

  • Deployed: Wednesday, February 22, 2017

  • Contributors: Rebecca Bilbro, Benjamin Bengfort

  • TSNEVisualizer for 2D projections of vectorized documents

  • FreqDistVisualizer for token frequency of text in a corpus

  • Added the user testing evaluation to the documentation

  • Created and host documentation there with RFD

  • Created a sample corpus and text examples notebook

  • Created a base class for text, TextVisualizer

  • Model selection tutorial using Mushroom Dataset

  • Created a text examples notebook but have not added to documentation.

Version 0.3.2

Hardened the Yellowbrick API to elevate the idea of a Visualizer to a first principle. This included reconciling shifts in the development of the preliminary versions to the new API, formalizing Visualizer methods like draw() and finalize(), and adding utilities that revolve around scikit-learn. To that end we also performed administrative tasks like refreshing the documentation and preparing the repository for more and varied open source contributions.

  • Tag: v0.3.2

  • Deployed: Friday, January 20, 2017

  • Contributors: Benjamin Bengfort, Rebecca Bilbro

  • Converted Mkdocs documentation to Sphinx documentation

  • Updated docstrings for all Visualizers and functions

  • Created a DataVisualizer base class for dataset visualization

  • Single call functions for simple visualizer interaction

  • Added yellowbrick specific color sequences and palettes and env handling

  • More robust examples with downloader from DDL host

  • Better axes handling in visualizer, matplotlib/sklearn integration

  • Added a finalize method to complete drawing before render

  • Improved testing on real data sets from examples

  • Bugfix: score visualizer renders in notebook but not in Python scripts.

  • Bugfix: tests updated to support new API

Hotfix 0.3.1

Hotfix to solve pip install issues with Yellowbrick.

  • Tag: v0.3.1

  • Deployed: Monday, October 10, 2016

  • Contributors: Benjamin Bengfort

    • Modified packaging and wheel for Python 2.7 and 3.5 compatibility

    • Modified deployment to PyPI and pip install ability

    • Fixed Travis-CI tests with the backend failures.

Version 0.3

This release marks a major change from the previous MVP releases as Yellowbrick moves towards direct integration with scikit-learn for visual diagnostics and steering of machine learning and could therefore be considered the first alpha release of the library. To that end we have created a Visualizer model which extends sklearn.base.BaseEstimator and can be used directly in the ML Pipeline. There are a number of visualizers that can be used throughout the model selection process, including for feature analysis, model selection, and hyperparameter tuning.

In this release specifically, we focused on visualizers in the data space for feature analysis and visualizers in the model space for scoring and evaluating models. Future releases will extend these base classes and add more functionality.

  • Tag: v0.3

  • Deployed: Sunday, October 9, 2016

  • Contributors: Benjamin Bengfort, Rebecca Bilbro, Marius van Niekerk

    • Created an API for visualization with machine learning: Visualizers that are BaseEstimators.

    • Created a class hierarchy for Visualizers throughout the ML process particularly feature analysis and model evaluation

    • Visualizer interface is draw method which can be called multiple times on data or model spaces and a poof method to finalize the figure and display or save to disk.

    • ScoreVisualizers wrap scikit-learn estimators and implement fit() and predict() (pass-throughs to the estimator) and also score which calls draw in order to visually score the estimator. If the estimator isn’t appropriate for the scoring method an exception is raised.

    • ROCAUC is a ScoreVisualizer that plots the receiver operating characteristic curve and displays the area under the curve score.

    • ClassificationReport is a ScoreVisualizer that renders the confusion matrix of a classifier as a heatmap.

    • PredictionError is a ScoreVisualizer that plots the actual vs. predicted values and the 45 degree accuracy line for regressors.

    • ResidualPlot is a ScoreVisualizer that plots the residuals (y - yhat) across the actual values (y) with the zero accuracy line for both train and test sets.

    • ClassBalance is a ScoreVisualizer that displays the support for each class as a bar plot.

    • FeatureVisualizers are scikit-learn Transformers that implement fit() and transform() and operate on the data space, calling draw to display instances.

    • ParallelCoordinates plots instances with class across each feature dimension as line segments across a horizontal space.

    • RadViz plots instances with class in a circular space where each feature dimension is an arc around the circumference and points are plotted relative to the weight of the feature.

    • Rank2D plots pairwise scores of features as a heatmap in the space [-1, 1] to show relative importance of features. Currently implemented ranking functions are Pearson correlation and covariance.

    • Coordinated and added palettes in the bgrmyck space and implemented a version of the Seaborn set_palette and set_color_codes functions as well as the ColorPalette object and other matplotlib.rc modifications.

    • Inherited Seaborn’s notebook context and whitegrid axes style but make them the default, don’t allow user to modify (if they’d like to, they’ll have to import Seaborn). This gives Yellowbrick a consistent look and feel without giving too much work to the user and prepares us for matplotlib 2.0.

    • Jupyter Notebook with Examples of all Visualizers and usage.

    Bug Fixes:
    • Fixed Travis-CI test failures with matplotlib.use(‘Agg’).

    • Fixed broken link to Quickstart on README

    • Refactor of the original API to the scikit-learn Visualizer API

Version 0.2

Intermediate steps towards a complete API for visualization. Preparatory stages for scikit-learn visual pipelines.

  • Tag: v0.2

  • Deployed: Sunday, September 4, 2016

  • Contributors: Benjamin Bengfort, Rebecca Bilbro, Patrick O’Melveny, Ellen Lowy, Laura Lorenz

    • Continued attempts to fix the Travis-CI Scipy install failure (broken tests)

    • Utility function: get the name of the model

    • Specified a class based API and the basic interface (render, draw, fit, predict, score)

    • Added more documentation, converted to Sphinx, autodoc, docstrings for viz methods, and a quickstart

    • How to contribute documentation, repo images etc.

    • Prediction error plot for regressors (mvp)

    • Residuals plot for regressors (mvp)

    • Basic style settings a la seaborn

    • ROC/AUC plot for classifiers (mvp)

    • Best fit functions for “select best”, linear, quadratic

    • Several Jupyter notebooks for examples and demonstrations

Version 0.1

Created the yellowbrick library MVP with two primary operations: a classification report heat map and a ROC/AUC curve model analysis for classifiers. This is the base package deployment for continuing yellowbrick development.

  • Tag: v0.1

  • Deployed: Wednesday, May 18, 2016

  • Contributors: Benjamin Bengfort, Rebecca Bilbro

    • Created the Anscombe quartet visualization example

    • Added DDL specific color maps and a stub for more style handling

    • Created crplot which visualizes the confusion matrix of a classifier

    • Created rocplot_compare which compares two classifiers using ROC/AUC metrics

    • Stub tests/stub documentation