evalml.pipelines.normalize_confusion_matrix

evalml.pipelines.normalize_confusion_matrix(conf_mat, normalize_method='true')

Normalizes a confusion matrix.

Parameters

• conf_mat (pd.DataFrame or np.array) – confusion matrix to normalize.

• 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

A normalized version of the input confusion matrix.

evalml.pipelines.confusion_matrix evalml.pipelines.graph_confusion_matrix