提交 a04450f7 编写于 作者: L lujun

fix bug for sequence_tagging_for_ner in python3, test=develop

上级 94e60c54
...@@ -30,7 +30,9 @@ def test(exe, chunk_evaluator, inference_program, test_data, test_fetch_list, ...@@ -30,7 +30,9 @@ def test(exe, chunk_evaluator, inference_program, test_data, test_fetch_list,
num_infer = np.array(rets[0]) num_infer = np.array(rets[0])
num_label = np.array(rets[1]) num_label = np.array(rets[1])
num_correct = np.array(rets[2]) num_correct = np.array(rets[2])
chunk_evaluator.update(num_infer[0], num_label[0], num_correct[0]) chunk_evaluator.update(num_infer[0].astype('int64'),
num_label[0].astype('int64'),
num_correct[0].astype('int64'))
return chunk_evaluator.eval() return chunk_evaluator.eval()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册