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

fix : load model test

parent b26847c5
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 3 additions and 3 deletions
image_ref/main.py
+ 3
− 3
View file @ 830d353e
......@@ -135,12 +135,12 @@ def run_duo(args):
plt.show()
plt.savefig('../output/training_plot_contrastive_{}.png'.format(args.positive_prop))
plt.savefig('output/training_plot_contrastive_{}.png'.format(args.positive_prop))
#load and evaluate best model
load_model(model, args.save_path)
make_prediction_duo(model,data_test_batch, '../output/confusion_matrix_contractive_{}_bis.png'.format(args.positive_prop),
'../output/confidence_matrix_contractive_{}_bis.png'.format(args.positive_prop))
make_prediction_duo(model,data_test_batch, 'output/confusion_matrix_contractive_{}_bis.png'.format(args.positive_prop),
'output/confidence_matrix_contractive_{}_bis.png'.format(args.positive_prop))
def make_prediction_duo(model, data, f_name, f_name2):
......
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