Spacy Ner定制训练和预训练模型的置信度得分
问题描述
我在Spacy的文件中看到,最近版本中推出了NER实体的置信度分数。我正在使用spacy==3.1.2。我尝试了以下代码来查找置信度分数,但得到了一个错误。此外,是否可以找到自定义模型和预先训练的模型的置信度分数。
代码
nlp = spacy.load("output/model-best")
test_data = 'Sample data here'
doc = nlp(test_data)
spans = doc.spans["spancat"] # SpanGroup
print(spans.attrs["scores"]) # list of numbers, span length as SpanGroup
错误
KeyError: 'spancat'
解决方案
实体分数仅在使用spancat组件时可用。我不确定您的代码应该做什么,但这不是您使用spancat组件的方式,您需要使用它训练管道。
预先训练的管道使用EntityRecognizer/NER组件,该组件不支持置信度分数。
相关文章