shuxinyin / NER-Pytorch-Chinese

Implemention of NER model on chinese dataset.
MIT License
64 stars 10 forks source link

bug? #5

Closed fxb392 closed 1 year ago

fxb392 commented 1 year ago
        if args.model_class in ['lebert-crf', 'bert-crf']:
            preds = model.crf.decode(logits, attention_mask).squeeze(0)
            preds = preds[:, 1:].tolist()  # 减去padding的[CLS]

评估函数172行,是不是少了对bert-lstm-crf的decoder解码 image

shuxinyin commented 1 year ago

这是两种解方式,crf-decode与argmax, 两种方式都是可以的。这里对应bert-lstm-crf确实应该用crf解码, 已添加。