From edb9f7b55a1a7163c91f915901a49f382ca4a548 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Wed, 2 Apr 2025 13:38:06 +0200
Subject: [PATCH] fix : make_prediction
---
image_ref/main.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/image_ref/main.py b/image_ref/main.py
index 3d983bc..12a696e 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -169,7 +169,7 @@ def make_prediction_duo(model, data, f_name, f_name2):
# constant for classes
# Build confusion matrix
- classes = data.dataset.dataset.classes
+ classes = data.dataset.classes
cf_matrix = confusion_matrix(y_true, y_pred)
confidence_matrix = np.zeros((n_class,n_class))
for i in range(n_class):
--
GitLab