我想在拥抱脸管道中使用“grouped_entities”来完成 ner 任务。但是这样做有问题。
我确实在 git 上查看了以下链接,但这没有帮助: https ://github.com/huggingface/transformers/pull/4987
我想在拥抱脸管道中使用“grouped_entities”来完成 ner 任务。但是这样做有问题。
我确实在 git 上查看了以下链接,但这没有帮助: https ://github.com/huggingface/transformers/pull/4987
我在变压器 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)