snapshot...

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