evalml.AutoClassificationSearch.plot.generate_confusion_matrix

AutoClassificationSearch.plot.generate_confusion_matrix(pipeline_id, fold_num=None, normalize=None)

Generate confusion matrix plot for a given pipeline using the data returned from get_confusion_matrix_data().

Parameters

• pipeline_id (int) – ID of pipeline to get confusion matrix data for

• fold_num (int) – Fold number of pipeline to get confusion matrix data for

• option ({'true', 'pred', 'all', None}) – Option to normalize over the rows (‘true’), columns (‘pred’) or all (‘all’) values. If option is None, returns original confusion matrix. Defaults to ‘true’.

Returns

plotly.Figure representing the confusion matrix plot generated