From 9e22d1c416ccfaecb2ef2bcde8d843956a634b3a Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Wed, 2 Apr 2025 16:38:01 +0200
Subject: [PATCH] fix : negative pair is always negative
---
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 3dda6c8..704f625 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -137,8 +137,8 @@ def run_duo(args):
#load and evaluate best model
load_model(model, args.save_path)
- make_prediction_duo(model,data_test_batch, 'output/confusion_matrix_contractive_{}.png'.format(args.prop),
- 'output/confidence_matrix_contractive_{}.png'.format(args.prop))
+ make_prediction_duo(model,data_test_batch, 'output/confusion_matrix_contractive_{}.png'.format(args.positive_prop),
+ 'output/confidence_matrix_contractive_{}.png'.format(args.positive_prop))
def make_prediction_duo(model, data, f_name, f_name2):
--
GitLab