Closed fxb392 closed 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解码
这是两种解方式,crf-decode与argmax, 两种方式都是可以的。这里对应bert-lstm-crf确实应该用crf解码, 已添加。
评估函数172行,是不是少了对bert-lstm-crf的decoder解码