f2io/ner-roles-openapi

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