yellowbrick

What it is and what it does

Yellowbrick wraps estimators with a Visualizer API that generates diagnostic plots for the model selection workflow. It implements visualizers for classification (confusion matrices, ROC curves, precision-recall plots), clustering (silhouette analysis, elbow curves), feature analysis (PCA projections, feature importance rankings), regression (residual plots, prediction error analysis), and text data (term frequency, t-SNE corpus plots). Each visualizer learns from data like a transformer or model, integrating into existing workflows.

The library's core strength is reducing the friction of exploratory model diagnostics—instead of writing matplotlib code to plot cross-validation scores, learning curves, or class balance, you instantiate a visualizer, fit it to your data, and call show(). It depends on matplotlib for rendering, scipy and numpy for computation, and scikit-learn for the estimator interface. The package is stable and production-ready but dormant; no active development has occurred since mid-2022.

Use it for

• Visualize confusion matrices and classification reports to diagnose precision, recall, and F1 imbalances across classes
• Plot learning curves to determine whether a model would benefit from more training data or reduced complexity
• Generate ROC and precision-recall curves to evaluate binary and multiclass classifier thresholds
• Identify optimal cluster count using silhouette scores or elbow detection in clustering workflows
• Rank feature importance and perform recursive feature elimination to guide feature engineering
• Inspect residuals and prediction errors in regression models to detect systematic model failures

Worth the install?

AI-flagged interpretation of the facts on this page. Verify before relying on it.

Install

yellowbrick on PyPI

Before you install

Low friction install with standard scientific Python stack dependencies (matplotlib, scipy, scikit-learn, numpy). Maintenance is dormant—last release was 2022-08-21 and no commits since 2025-02-19, so expect no active bug fixes or feature development, though the codebase remains stable.

License in practice

Apache 2 permissive license allows commercial and private use without restriction, making it safe for most projects.

Quickstart

pip install yellowbrick

from yellowbrick.classifier import ConfusionMatrix

visualizer = ConfusionMatrix(estimator)
visualizer.fit(X_train, y_train)
visualizer.score(X_test, y_test)
visualizer.show()

Package facts

Evidence: yellowbrick-1.5-py3-none-any.whl

Tags

More Software Development packages

See also scikit-plot · missingno · treescope · dtreeviz · azureml-train-core · pm4py · scikit-learn · spark-sklearn · scikit-multilearn · feature-engine