Merge pull request namisan#100 from namisan/fix_calc_metrics

fix small bug in predict.py
atherosai · Oct 3, 2019 · fc80aee · fc80aee
2 parents a579971 + afc3399
commit fc80aee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/predict.py b/predict.py
@@ -31,7 +31,7 @@ def eval_model(model, data, metric_meta, use_cuda=True, with_label=True):
         scores.extend(score)
         ids.extend(batch_meta['uids'])
     if with_label:
-        metrics = calc_metrics(metric_meta, golds, predictions, scores, ids)
+        metrics = calc_metrics(metric_meta, golds, predictions, scores)
     return metrics, predictions, scores, golds, ids