Skip to content
Snippets Groups Projects
Commit 3900c13f authored by Schneider Leo's avatar Schneider Leo
Browse files

fix : output f_name

parent c4b40467
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 2 additions and 2 deletions
image_ref/main.py
+ 2
− 2
View file @ 3900c13f
...@@ -107,10 +107,10 @@ def run_duo(args): ...@@ -107,10 +107,10 @@ def run_duo(args):
plt.ylim(0, 1.05) plt.ylim(0, 1.05)
plt.show() 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 and evaluate best model
load_model(model, args.save_path) 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): def make_prediction_duo(model, data, f_name):
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment