diff --git a/modules/util/plot_cm.py b/modules/util/plot_cm.py index eafcb4c6b4b3fd1cdfc5f7f61fd395a704df5e49..34eabf6605e76e41284921a134224ae24fda8712 100644 --- a/modules/util/plot_cm.py +++ b/modules/util/plot_cm.py @@ -11,6 +11,7 @@ import matplotlib.pyplot as plt def plot_confusion_matrix_values(correct_labels, predict_labels, labels, title='Confusion matrix', filename='confusion_matrix', normalize=False): cm = confusion_matrix(correct_labels, predict_labels) + return cm def plot_confusion_matrix(cm, labels, title='Confusion matrix', filename = 'confusion_matrix', normalize=False):