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

fix : out file name

parent a9e0ea7f
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 1 addition and 1 deletion
image_ref/main.py
+ 1
− 1
View file @ e24ccbb9
...@@ -184,7 +184,7 @@ def run_duo(args): ...@@ -184,7 +184,7 @@ def run_duo(args):
load_model(model, args.save_path) load_model(model, args.save_path)
if args.dataset_test_dir is not None : if args.dataset_test_dir is not None :
make_prediction_duo(model, data_test_batch,args.base_out+'_confusion_matrix_test.png', make_prediction_duo(model, data_test_batch,args.base_out+'_confusion_matrix_test.png',
args.base_out+'confidence_matrix_test.png') args.base_out+'_confidence_matrix_test.png')
make_prediction_duo(model, data_val_batch,args.base_out+'_confusion_matrix_val.png', make_prediction_duo(model, data_val_batch,args.base_out+'_confusion_matrix_val.png',
args.base_out+'_confidence_matrix_val.png') args.base_out+'_confidence_matrix_val.png')
......
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