debug printout

francesco-vaselli · May 21, 2024 · 9a80aca · 9a80aca
1 parent 92b3468
commit 9a80aca
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/utils.py b/src/utils.py
@@ -490,6 +490,7 @@ def plot_confusion_matrix1(cm, class_names):
 
 def check_classification3(true, pred):
     # Assuming true and pred have shape [batch_size, 3]
+    print("true shape:", true.shape)
     true_label = np.argmax(true, axis=1)
     pred_label = np.argmax(pred, axis=1)
     print("true_label shape:", true_label.shape)