From 3900c13f4502f9dfae0e88fe872266beedc92160 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Tue, 1 Apr 2025 14:52:45 +0200
Subject: [PATCH] fix : output f_name
---
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 ff56909..e2dbdeb 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -107,10 +107,10 @@ def run_duo(args):
plt.ylim(0, 1.05)
plt.show()
- plt.savefig('output/training_plot_contrastive_noise_{}_lr_{}_model_{}_{}.png'.format(args.noise_threshold,args.lr,args.model,args.model_type))
+ plt.savefig('output/training_plot_contrastive_noise_{}_lr_{}_model_{}.png'.format(args.noise_threshold,args.lr,args.model))
#load and evaluate best model
load_model(model, args.save_path)
- make_prediction_duo(model,data_test, 'output/confusion_matrix_contractive_noise_{}_lr_{}_model_{}_{}.png'.format(args.noise_threshold,args.lr,args.model,args.model_type))
+ make_prediction_duo(model,data_test, 'output/confusion_matrix_contractive_noise_{}_lr_{}_model_{}_{}.png'.format(args.noise_threshold,args.lr,args.model))
def make_prediction_duo(model, data, f_name):
--
GitLab