From 52e0d66e939ab0e59ccbd591bed5ea68d424512e Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Wed, 12 Mar 2025 18:23:33 +0100
Subject: [PATCH] fix confusion matrix
---
main.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/main.py b/main.py
index a4074a2..a9aa6e4 100644
--- a/main.py
+++ b/main.py
@@ -236,9 +236,9 @@ def make_prediction_duo(model, data, f_name):
# Build confusion matrix
cf_matrix = confusion_matrix(y_true, y_pred)
- df_cm = pd.DataFrame(cf_matrix / np.sum(cf_matrix, axis=1)[:, None], index=[i for i in classes],
+ df_cm = pd.DataFrame(cf_matrix[:, None], index=[i for i in classes],
columns=[i for i in classes])
- plt.figure(figsize=(12, 7))
+ plt.figure(figsize=(14, 9))
sn.heatmap(df_cm, annot=True)
plt.savefig(f_name)
--
GitLab