diff --git a/image_ref/main.py b/image_ref/main.py index 35fd390c66ccacb2109420a670905ea4e9915591..0d109924a702ab54673efe14e823f09a5b9487cb 100644 --- a/image_ref/main.py +++ b/image_ref/main.py @@ -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.positive_prop), - 'output/confidence_matrix_contractive_{}.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):