From 488b3c92faa049a6d8bdd125d06124a856aa37f0 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Fri, 21 Mar 2025 09:36:02 +0100
Subject: [PATCH] debugging confusion matrix
---
main.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/main.py b/main.py
index d2227c6..29ae275 100644
--- a/main.py
+++ b/main.py
@@ -233,7 +233,7 @@ def make_prediction_duo(model, data, f_name):
# constant for classes
classes = data.dataset.dataset.classes
-
+ print('Prediction made')
# Build confusion matrix
cf_matrix = confusion_matrix(y_true, y_pred)
df_cm = pd.DataFrame(cf_matrix[:, None], index=[i for i in classes],
--
GitLab