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