Simple finetuned RoBERTA 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", ]
- Downloads last month
- 1