def METHOD_NAME(self):
    """
    Returns the plots in a dict format.
        {
            'confusion_matrix': confusion matrix figure,
        }
    """
    display = metrics.ConfusionMatrixDisplay(confusion_matrix=self.confusion_matrix, display_labels=[False, True])
    return {'confusion_matrix': display.plot().figure_}