ljynlp/W2NER

预测结果中的实体并不是按顺序出现的

zhYua opened this issue · 6 comments

zhYua commented

我用W2NER在自己数据集上进行预测后，出来的实体结果顺序并不是按照原始文本中的顺序，比如我的文本是ABC，但是预测的实体顺序是ACB这样，请问这个顺序是需要自己后处理调么，还是在inference的时候可以设置，或者是我 预测有问题

ljynlp commented

可以举一个具体的例子吗？

zhYua commented

比如：超逸三组调味盒，我想要的结果是超逸：品牌，三组：规格，调味盒：商品 但是现在预测出来的json文件中，不是按照超逸、三组、调味盒的顺序出现的，而是乱序
image

ljynlp commented

我们在解码的时候为了处理可能存在的重复抽取的情况，使用了set对结果进行了去重，因此不能保证结果的顺序，如果需要可以自行根据实体的位置进行排序。

zhYua commented

那请问解码时候有输出实体头尾位置的这个返回值么

ljynlp commented

输出结果并没有实体的位置信息，具体可以在utils.py中的decode()中进行修改。

  • 👍1
zhYua commented

好的，谢谢