evalml.pipelines.confusion_matrix

evalml.pipelines.confusion_matrix(y_true, y_predicted, normalize_method='true')

Confusion matrix for binary and multiclass classification.

Parameters

• y_true (pd.Series or np.array) – true binary labels.

• y_pred (pd.Series or np.array) – predictions from a binary classifier.

• normalize_method ({'true', 'pred', 'all'}) – Normalization method. Supported options are: ‘true’ to normalize by row, ‘pred’ to normalize by column, or ‘all’ to normalize by all values. Defaults to ‘true’.

Returns

confusion matrix

Return type

np.array