From 77519ed7434524067148b121ca45038d6a1c9714 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Fri, 21 Mar 2025 10:17:08 +0100
Subject: [PATCH] debugging confusion matrix
---
config/config.py | 2 +-
main.py | 1 +
2 files changed, 2 insertions(+), 1 deletion(-)
diff --git a/config/config.py b/config/config.py
index 406b5af..789f0bc 100644
--- a/config/config.py
+++ b/config/config.py
@@ -4,7 +4,7 @@ import argparse
def load_args():
parser = argparse.ArgumentParser()
- parser.add_argument('--epoches', type=int, default=100)
+ parser.add_argument('--epoches', type=int, default=10)
parser.add_argument('--save_inter', type=int, default=50)
parser.add_argument('--eval_inter', type=int, default=1)
parser.add_argument('--noise_threshold', type=int, default=0)
diff --git a/main.py b/main.py
index ff27a40..6ef0753 100644
--- a/main.py
+++ b/main.py
@@ -235,6 +235,7 @@ def make_prediction_duo(model, data, f_name):
classes = data.dataset.dataset.classes
print('Prediction made')
# Build confusion matrix
+ print(len(y_true),len(y_pred))
cf_matrix = confusion_matrix(y_true, y_pred)
print('CM made')
df_cm = pd.DataFrame(cf_matrix[:, None], index=[i for i in classes],
--
GitLab