From d4ad97576780ce3c1a3b7eaa315bbb60c4e21ea4 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Mon, 14 Apr 2025 11:06:33 +0200
Subject: [PATCH] fix : typo
---
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 cbb8986e..a06cac93 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -169,11 +169,11 @@ def run_duo(args):
plt.tight_layout()
plt.show()
- plt.savefig('output/training_plot_contrastive_{}.png'.format(args.positive_prop))
+ plt.savefig(args.base_out+'_training_plot.png')
# load and evaluate best model
load_model(model, args.save_path)
- if args.args.dataset_test_dir is not None :
+ if args.dataset_test_dir is not None :
make_prediction_duo(model, data_test_batch,args.base_out+'_confusion_matrix_test.png',
args.base_out+'confidence_matrix_.png')
--
GitLab