allenai / allenai/scibert

Error while running Named Entity Recognition

未关闭
#97 1 条评论 1 个 reaction 已指派 0 人 在 GitHub 查看
主要语言
Python
星标
1.7k
派生
232
PR 合并指标
30 天内没有已合并 PR

描述

I tried using SciBERT for NER using the following command-

from transformers import *

tokenizer = AutoTokenizer.from_pretrained('allenai/scibert_scivocab_uncased')
model = AutoModel.from_pretrained('allenai/scibert_scivocab_uncased')

nlp = pipeline('ner',model = model,tokenizer=tokenizer)
nlp('Clinical features of culture-proven Mycoplasma pneumoniae infections at King Abdulaziz University Hospital, Jeddah, Saudi Arabia')

While running it on a sample sentence, I get the following error-
Traceback (most recent call last):
File "", line 1, in
File "/python3.6/site-packages/transformers/pipelines.py", line 927, in __call__
for idx, label_idx in enumerate(labels_idx)
File "/python3.6/site-packages/transformers/pipelines.py", line 928, in
if self.model.config.id2label[label_idx] not in self.ignore_labels
KeyError: 422

贡献指南

这个仓库没有索引到贡献指南

评估

这个 Issue 还没有评估数据。

把新 issue 发到你的邮箱

精选适合新手参与的 GitHub issue 摘要。