From 830d353e7f0c8847102a364c94b11f8f942b846c Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Fri, 4 Apr 2025 15:37:58 +0200
Subject: [PATCH] fix : load model test
---
image_ref/main.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/image_ref/main.py b/image_ref/main.py
index 8e33af8..017daec 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -135,12 +135,12 @@ def run_duo(args):
plt.show()
- plt.savefig('../output/training_plot_contrastive_{}.png'.format(args.positive_prop))
+ plt.savefig('output/training_plot_contrastive_{}.png'.format(args.positive_prop))
#load and evaluate best model
load_model(model, args.save_path)
- make_prediction_duo(model,data_test_batch, '../output/confusion_matrix_contractive_{}_bis.png'.format(args.positive_prop),
- '../output/confidence_matrix_contractive_{}_bis.png'.format(args.positive_prop))
+ make_prediction_duo(model,data_test_batch, 'output/confusion_matrix_contractive_{}_bis.png'.format(args.positive_prop),
+ 'output/confidence_matrix_contractive_{}_bis.png'.format(args.positive_prop))
def make_prediction_duo(model, data, f_name, f_name2):
--
GitLab