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 |