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