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

fix : negative pair is always negative

parent b1d42708
No related branches found
No related tags found
No related merge requests found
......@@ -137,8 +137,8 @@ def run_duo(args):
#load and evaluate best model
load_model(model, args.save_path)
make_prediction_duo(model,data_test_batch, 'output/confusion_matrix_contractive_{}.png'.format(args.prop),
'output/confidence_matrix_contractive_{}.png'.format(args.prop))
make_prediction_duo(model,data_test_batch, 'output/confusion_matrix_contractive_{}.png'.format(args.positive_prop),
'output/confidence_matrix_contractive_{}.png'.format(args.positive_prop))
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