evalml.AutoClassificationSearch.plot.get_confusion_matrix_data

AutoClassificationSearch.plot.get_confusion_matrix_data(pipeline_id, normalize=None)

Gets data that can be used to create a confusion matrix plot.

Parameters

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

• normalize ({'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

List containing information used to generate a confusion matrix plot. Each element in the list contains the confusion matrix data for that fold.