spaCy中EntityRecognizer无法调用add_label()方法的问题

问题描述 #

尝试通过自定义示例更新spaCy的实体识别器，获取Named Entity Recognizer管道组件实例后调用add_label()方法，代码如下：

import spacy

nlp = spacy.load('en_core_web_sm')
ner = nlp.get_pipe('ner')
ner.add_label()

操作时收到提示：Cannot find reference 'add_label' in 'Pipe | Pipe'，无法调用该方法。原本以为add_label()在spaCy v3中已被弃用，但查阅官方文档确认该方法并未被弃用，且当前管道确实包含ner组件：

print(nlp.pipe_names)
# 输出: ['tok2vec', 'tagger', 'parser', 'attribute_ruler', 'lemmatizer', 'ner']

环境信息 #

• spaCy版本：3.4.3
• 预训练模型：en_core_web_sm（3.4.1）
• Python版本：3.10.4

问题原因与解决 #

更新：经测试，该问题由PyCharm（版本17.0.5+1-b653.14 x86_64）导致，在VS Code中无法复现。

这是PyCharm静态代码分析的误报，并非spaCy API本身存在问题：

• spaCy v3.x版本的EntityRecognizer确实保留了add_label()方法，用于添加新的实体标签，代码在实际运行时是可以正常执行的。
• 旧版本PyCharm对spaCy的类型提示支持不完善，无法正确识别nlp.get_pipe('ner')返回的EntityRecognizer实例的方法，从而触发错误提示。

解决建议 #

• 更新PyCharm到最新版本，新版本对spaCy的类型支持更完善，可解决该误报问题。
• 临时解决方案：给ner变量添加类型注解，明确其类型，帮助PyCharm识别方法：

import spacy
from spacy.pipeline import EntityRecognizer

nlp = spacy.load('en_core_web_sm')
ner: EntityRecognizer = nlp.get_pipe('ner')
ner.add_label("NEW_LABEL")  # 替换为你要添加的实体标签

内容的提问来源于stack exchange，提问作者paswes