roberta-nerc / README.md
pawlo2013's picture
Update README.md
e2b1765 verified
---
datasets:
- djagatiya/ner-ontonotes-v5-eng-v4
---
Simple finetuned <a href="https://huggingface.co/FacebookAI/roberta-base" target="_blank"> RoBERTA</a> for NERC(Named Entity Recognitiona and Classification) task.
The model was finetuned for the Text Mining course at Vrije Universieit Amsterdam in 2024.
Classes:
entity_list = [
"B-CARDINAL",
"I-CARDINAL",
"B-DATE",
"I-DATE",
"B-EVENT",
"I-EVENT",
"B-FAC",
"I-FAC",
"B-GPE",
"I-GPE",
"B-LANGUAGE",
"I-LANGUAGE",
"B-LAW",
"I-LAW",
"B-LOC",
"I-LOC",
"B-MONEY",
"I-MONEY",
"B-NORP",
"I-NORP",
"B-ORDINAL",
"I-ORDINAL",
"B-ORG",
"I-ORG",
"B-PERCENT",
"I-PERCENT",
"B-PERSON",
"I-PERSON",
"B-PRODUCT",
"I-PRODUCT",
"B-QUANTITY",
"I-QUANTITY",
"B-TIME",
"I-TIME",
"B-WORK_OF_ART",
"I-WORK_OF_ART",
"O",
]