pipe.model.config.id2label

#2
by wolfgangblack - opened

Hello,

I called down this model to experiment with the ner capabilities and found that it did not have true labels.

When I run

from transformers import pipeline

pipe = pipeline("token-classification", model="shibing624/bert4ner-base-uncased")

pipe.model.config.id2label

i get

{0: 'LABEL_0',
 1: 'LABEL_1',
 2: 'LABEL_2',
 3: 'LABEL_3',
 4: 'LABEL_4',
 5: 'LABEL_5',
 6: 'LABEL_6',
 7: 'LABEL_7',
 8: 'LABEL_8',
 9: 'LABEL_9',
 10: 'LABEL_10',
 11: 'LABEL_11',
 12: 'LABEL_12',
 13: 'LABEL_13',
 14: 'LABEL_14',
 15: 'LABEL_15',
 16: 'LABEL_16'}
shibing624 changed discussion status to closed

Sign up or log in to comment