From 5710da1977f1d0135e867d65de99d38a5ae2b229 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Wed, 2 Apr 2025 16:07:32 +0200
Subject: [PATCH] change : change training and testing split
---
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 9ac9928..3dda6c8 100644
--- a/image_ref/main.py
+++ b/image_ref/main.py
@@ -133,12 +133,12 @@ def run_duo(args):
plt.show()
- plt.savefig('output/training_plot_contrastive_noise_{}_lr_{}_model_{}.png'.format(args.noise_threshold,args.lr,args.model))
+ plt.savefig('output/training_plot_contrastive_{}.png'.format(args.prop))
#load and evaluate best model
load_model(model, args.save_path)
- make_prediction_duo(model,data_test_batch, 'output/confusion_matrix_contractive_noise_{}_lr_{}_model_{}.png'.format(args.noise_threshold,args.lr,args.model),
- 'output/confidence_matrix_contractive_noise_{}_lr_{}_model_{}.png'.format(args.noise_threshold,args.lr,args.model))
+ make_prediction_duo(model,data_test_batch, 'output/confusion_matrix_contractive_{}.png'.format(args.prop),
+ 'output/confidence_matrix_contractive_{}.png'.format(args.prop))
def make_prediction_duo(model, data, f_name, f_name2):
--
GitLab