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

change : change training and testing split

parent f557ebbe
No related branches found
No related tags found
No related merge requests found
...@@ -133,12 +133,12 @@ def run_duo(args): ...@@ -133,12 +133,12 @@ def run_duo(args):
plt.show() 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 and evaluate best model
load_model(model, args.save_path) 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), make_prediction_duo(model,data_test_batch, 'output/confusion_matrix_contractive_{}.png'.format(args.prop),
'output/confidence_matrix_contractive_noise_{}_lr_{}_model_{}.png'.format(args.noise_threshold,args.lr,args.model)) 'output/confidence_matrix_contractive_{}.png'.format(args.prop))
def make_prediction_duo(model, data, f_name, f_name2): def make_prediction_duo(model, data, f_name, f_name2):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment