Class Prediction Error

The class prediction error chart provides a way to quickly understand how good your classifier is at predicting the right classes.

from sklearn.datasets import make_classification

# Create classification dataset
X, y = make_classification(
    n_samples=1000, n_classes=5, n_informative=3, n_clusters_per_class=1

# Name the classes
classes = ['apple', 'kiwi', 'pear', 'banana', 'orange']

# Perform 80/20 training/test split
X_train, X_test, y_train, y_test = tts(
    X, y, test_size=0.20, random_state=42
# Instantiate the classification model and visualizer
visualizer = ClassPredictionError(
    RandomForestClassifier(), classes=classes

# Fit the training data to the visualizer, y_train)

# Evaluate the model on the test data
visualizer.score(X_test, y_test)

# Draw visualization
g = visualizer.poof()

API Reference

Class balance visualizer for showing per-class support.

class yellowbrick.classifier.class_balance.ClassPredictionError(model, ax=None, classes=None, **kwargs)[source]

Bases: yellowbrick.classifier.base.ClassificationScoreVisualizer

Class Prediction Error chart that shows the support for each class in the fitted classification model displayed as a stacked bar. Each bar is segmented to show the distribution of predicted classes for each class. It is initialized with a fitted model and generates a class prediction error chart on draw.


ax: axes

the axis to plot the figure on.

model: estimator

Scikit-Learn estimator object. Should be an instance of a classifier, else __init__() will raise an exception.

classes: list

A list of class names for the legend. If classes is None and a y value is passed to fit then the classes are selected from the target vector.

kwargs: dict

Keyword arguments passed to the super class. Here, used to colorize the bars in the histogram.



These parameters can be influenced later on in the visualization

process, but can and should be set as early as possible.


Renders the class prediction error across the axis. Returns ——- ax : the axis with the plotted figure


Finalize executes any subclass-specific axes finalization steps. The user calls poof and poof calls finalize. Parameters ———- kwargs: generic keyword arguments.

score(X, y, **kwargs)[source]

Generates a 2D array where each row is the count of the predicted classes and each column is the true class


X : ndarray or DataFrame of shape n x m

A matrix of n instances with m features

y : ndarray or Series of length n

An array or series of target or class values


ax : the axis with the plotted figure