0

我想在拥抱脸管道中使用“grouped_entities”来完成 ner 任务。但是这样做有问题。

我确实在 git 上查看了以下链接,但这没有帮助: https ://github.com/huggingface/transformers/pull/4987

4

1 回答 1

0

我在变压器 v4.0.0 中得到了非常直接的答案。以前我使用的是旧版本的变压器包。例子:

from transformers import AutoTokenizer, 
 AutoModelForTokenClassification,TokenClassificationPipeline

 from transformers import pipeline 

tokenizer = AutoTokenizer.from_pretrained("fine_tune_model_path")
model = AutoModelForTokenClassification.from_pretrained("fine_tune_model_path")
nlp=pipeline(task='ner',model=model,
 tokenizer=tokenizer,grouped_entities=True)
于 2020-12-18T11:16:55.543 回答