Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
Convert the numbers to their label names to find out what the entities are:
label_list = wnut["train"].features[f"ner_tags"].feature.names
label_list
[
"O",
"B-corporation",
"I-corporation",
"B-creative-work",
"I-creative-work",
"B-group",
"I-group",
"B-location",
"I-location",
"B-person",
"I-person",
"B-product",
"I-product",
]
The letter that prefixes each ner_tag indicates the token position of the entity:
B- indicates the beginning of an entity.