evalml.pipelines.confusion_matrix

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

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

evalml.pipelines.graph_roc_curve evalml.pipelines.normalize_confusion_matrix