diff --git a/modules/util/plot_cm.py b/modules/util/plot_cm.py index 34eabf6605e76e41284921a134224ae24fda8712..64cfebe37dafb6761db3fac95a064c202f5b7398 100644 --- a/modules/util/plot_cm.py +++ b/modules/util/plot_cm.py @@ -9,7 +9,7 @@ from sklearn.metrics import confusion_matrix import matplotlib.pyplot as plt -def plot_confusion_matrix_values(correct_labels, predict_labels, labels, title='Confusion matrix', filename='confusion_matrix', normalize=False): +def confusion_matrix_values(correct_labels, predict_labels): cm = confusion_matrix(correct_labels, predict_labels) return cm