Browse Source

update CMRC2018Metric测试

tags/v0.5.5
yh_cc 4 years ago
parent
commit
faec3245d4
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      test/core/test_metrics.py

+ 2
- 1
test/core/test_metrics.py View File

@@ -430,7 +430,8 @@ class TestCMRC2018Metric(unittest.TestCase):
pred_start[1, 1] = 1000 # 取出234 pred_start[1, 1] = 1000 # 取出234
pred_end[1, 3] = 1000 pred_end[1, 3] = 1000


metric.evaluate(answers, raw_chars, context_len, pred_start, pred_end)
metric.evaluate(answers=answers, raw_chars=raw_chars, pred_start=pred_start,
pred_end=pred_end, context_len=context_len)


eval_res = metric.get_metric() eval_res = metric.get_metric()
self.assertDictEqual(eval_res, {'f1': 70.0, 'em': 50.0}) self.assertDictEqual(eval_res, {'f1': 70.0, 'em': 50.0})


Loading…
Cancel
Save