ner-roles-openapi: model fine-tuned from distilbert-base-uncased for NER task.
Introduction
[ner-roles-openapi] is a NER model that was fine-tuned from distilbert-base-uncased on synthetic dataset based SCRUM user story. Model was trained on synthetic dataset (~80 sentences). Model was validated on openapi/swagger data and overperformed other models on this type of data specifically.
Training data
Training data was classified as follow:
Abbreviation | Description |
---|---|
O | Outside of a named entity |
ENTITY | Entity provided by OpenAPI |
ACTION | Action required by a user/role |
ROLE | Role predefined to use APIs |