Merge pull request namisan#99 from namisan/fix_calc_metrics

fix small bug in calc metric
donglixp · Oct 3, 2019 · a579971 · a579971
2 parents 3154388 + 7d2689c
commit a579971
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/calc_metrics.py b/calc_metrics.py
@@ -39,10 +39,10 @@ def generate_golds_predictions_scores(sample_id_2_pred_score_seg_dic, sample_obj
 n_class = task_defs.n_class_map[task]
 sample_id_2_pred_score_seg_dic = load_score_file(args.score, n_class)
 
-data_format = task_defs.data_format_map[task]
+data_type = task_defs.data_type_map[task]
 task_type = task_defs.task_type_map[task]
 label_mapper = task_defs.global_map.get(task, None)
-sample_objs = load_data(args.std_input, data_format, task_type, label_mapper)
+sample_objs = load_data(args.std_input, data_type, task_type, label_mapper)
 
 golds, predictions, scores = generate_golds_predictions_scores(sample_id_2_pred_score_seg_dic, sample_objs)