z814081807 / DeepNER

天池中药说明书实体识别挑战冠军方案;中文命名实体识别;NER; BERT-CRF & BERT-SPAN & BERT-MRC;Pytorch
922 stars 229 forks source link

训练和预测不一致问题 #30

Closed kevinuserdd closed 3 years ago

kevinuserdd commented 3 years ago

mrc方法中的问题

  1. 训练时候构造query+context作为输入; 但是你预测时候就是context作为输入;
  2. 预测代码np.argmax()是多分类吧,那你一句话里面只能出现一个实体吗?
z814081807 commented 3 years ago

mrc方法中的问题

  1. 训练时候构造query+context作为输入; 但是你预测时候就是context作为输入;
  2. 预测代码np.argmax()是多分类吧,那你一句话里面只能出现一个实体吗?

预测的时候需要补充query 13类预测13次 我代码里没上传mrc单独预测的代码;至于为什么是argmax 请仔细看mrc方法的原理