From 7d2689ca8be519ee03b03485ea3ada282405c70b Mon Sep 17 00:00:00 2001 From: Anselm Wang Date: Tue, 1 Oct 2019 16:49:24 +0000 Subject: [PATCH] fix small bug in calc metric --- calc_metrics.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/calc_metrics.py b/calc_metrics.py index 4a53fa6a..82f79096 100644 --- 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)