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'}
use it : https://huggingface.co/shibing624/bert4ner-base-uncased#usage-huggingface-transformers , not support pipeline.
shibing624
changed discussion status to
closed