- 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.
- 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 withpredict_proba
anddecision_function
methods are handling. Abinary
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.
- Added Q-Q plot as side-by-side option to the
- Minor Changes:
- Allow users to specify colors for the
PrecisionRecallCurve
. - Update
ClassificationScoreVisualizer
base class to have aclass_colors_
learned attribute instead of acolors
property; additional polishing of multi-class colors inPrecisionRecallCurve
,ROCAUC
, andClassPredictionError
. - Update
KElbowVisualizer
fit method and quick method to allow passingsample_weight
parameter through the visualizer. - Enhancements to classification documentation to better discuss precision and recall and to diagnose with
PrecisionRecallCurve
andClassificationReport
visualizers. - Improvements to
CooksDistance
visualizer documentation. - Corrected
KElbowVisualizer
label and legend formatting. - Typo fixes to
ROCAUC
documentation, labels, and legend. Typo fix toManifold
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 movePredictionError
to its own module,yellowbrick.regressor.prediction_error
. - Update tests to use
SVC
instead ofLinearSVC
and correctKMeans
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
MANIFEST.in
. - 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.
- Allow users to specify colors for the
- 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.
- Tag: v1.0.1
- Deployed Sunday, October 6, 2019
- Contributors: Benjamin Bengfort, Rebecca Bilbro, Kristen McIntyre
Warning
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:
- Changes
poof()
toshow()
. - 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.
- Changes
- 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
Warning
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 toManifold
. - 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.target
,yellowbrick.features
, andyellowbrick.model_selection
API. - Restructured regressor and classifier API.
- New
- 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
README.md
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
withlegend
andcolor
parameter, while also move labels to the y-axis. - Extended
FeatureImportances
docs/tests for stacked importances. - Documented the
yellowbrick.download
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 otherax_bar
visualizers. - Updated
Manifold
to extendProjectionVisualizer
. - Check if an estimator is already fitted before calling
fit
method. - Ensure
poof
returnsax
.
- Updated
- 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
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
- 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/multiclassROCAUC
construction. - Rename tests/random.py to tests/rand.py to prevent NumPy errors.
- Improves
ROCAUC
,KElbowVisualizer
, andSilhouetteVisualizer
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.
- Removed hardcoding of
- 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
- Added Support to
- 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
- 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
Changes:
- New Feature! Manifold visualizers implement high-dimensional visualization for non-linear structural feature analysis.
- New Feature! There is now a
model_selection
module withLearningCurve
andValidationCurve
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 theMultipleVisualizer
that creates axes for each visualizer usingplt.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 toyellowbrick.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 asDiscriminationThreshold
, 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 andDecisionBoundaries
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 theVisualizerGrid
test cases. - Fix / update tests in
tests.test_classifier.test_class_prediction_error.py
to remove hardcoded data.
- Adds
- Deprecation Warnings:
ScatterPlotVisualizer
is being moved to contrib in 0.8DecisionBoundaryVisualizer
is being moved to contrib in 0.8ThreshViz
is renamed toDiscriminationThreshold
.
NOTE: These deprecation warnings originally mentioned deprecation in 0.7, but their life was extended by an additional version.
- 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
- Changes:
- New Feature! The
FeatureImportances
Visualizer 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
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
, andRank2D
. - Documentation updates, including discussion of Image Comparison Tests for contributors.
- New Feature! The
- 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 ofnp.tolist
to avoid NumPy deprecation warning. DataVisualizer
updated to removenp.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
- Fixes the
- Deprecation Warnings:
ScatterPlotVisualizer
is being moved to contrib in 0.7DecisionBoundaryVisualizer
is being moved to contrib in 0.7
- Tag: v0.5
- Deployed: Wednesday, August 9, 2017
- Contributors: Benjamin Bengfort, Rebecca Bilbro, Nathan Danielsen, Carlo Morales, Jim Stearns, Phillip Schafer, Jason Keung
- Changes:
- 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
- Added
- Bug Fixes:
- Repairs to
PCA
,RadViz
,FreqDist
unit tests - Repair to matplotlib version check in
JointPlotVisualizer
- Repairs to
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
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
- Changes:
- 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
withmodel_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
andcalinski_harabaz_score
computations and visualizations to KMeans visualizer. - Replaced the
self.ax
property on all of the individualdraw
methods with a new property on theVisualizer
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
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
- Changes:
- 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
- Part of speech tags visualizer --
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
- Changes:
TSNEVisualizer
for 2D projections of vectorized documentsFreqDistVisualizer
for token frequency of text in a corpus- Added the user testing evaluation to the documentation
- Created scikit-yb.org 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.
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
- Changes:
- 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 to solve pip install issues with Yellowbrick.
- Tag: v0.3.1
- Deployed: Monday, October 10, 2016
- Contributors: Benjamin Bengfort
- Changes:
- 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.
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
- Enhancements:
- 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 implementfit()
andpredict()
(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 aScoreVisualizer
that plots the receiver operating characteristic curve and displays the area under the curve score.ClassificationReport
is aScoreVisualizer
that renders the confusion matrix of a classifier as a heatmap.PredictionError
is aScoreVisualizer
that plots the actual vs. predicted values and the 45 degree accuracy line for regressors.ResidualPlot
is aScoreVisualizer
that plots the residuals (y - yhat) across the actual values (y) with the zero accuracy line for both train and test sets.ClassBalance
is aScoreVisualizer
that displays the support for each class as a bar plot.FeatureVisualizers
are scikit-learn Transformers that implementfit()
andtransform()
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.
- Created an API for visualization with machine learning: Visualizers that are
- 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
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
- Changes:
- 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
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
- Changes:
- 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