From 3b70076614ce7f93e1acf556375b74c790e8442c Mon Sep 17 00:00:00 2001 From: Schneider Leo <leo.schneider@etu.ec-lyon.fr> Date: Mon, 31 Mar 2025 14:45:23 +0200 Subject: [PATCH] fix : file path for jz --- 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 27565f2..738ea39 100644 --- a/image_ref/main.py +++ b/image_ref/main.py @@ -232,10 +232,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,args.model_type)) #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,args.model_type)) def make_prediction_duo(model, data, f_name): -- GitLab