From 609825a30c9506714d33ad973aa08e1446153691 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Wed, 2 Apr 2025 13:41:49 +0200
Subject: [PATCH] fix : make_prediction
---
image_ref/main.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/image_ref/main.py b/image_ref/main.py
index 12a696e..767cce8 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -175,8 +175,8 @@ def make_prediction_duo(model, data, f_name, f_name2):
for i in range(n_class):
confidence_matrix[i]=np.mean(confidence_pred_list[i],axis=0)
- df_cm = pd.DataFrame(cf_matrix / np.sum(cf_matrix, axis=1)[:, None], index=[i for i in range(2)],
- columns=['True','False'])
+ df_cm = pd.DataFrame(cf_matrix / np.sum(cf_matrix, axis=1)[:, None], index=[i for i in classes],
+ columns=[i for i in classes])
print('Saving Confusion Matrix')
plt.clf()
plt.figure(figsize=(14, 9))
--
GitLab