55+ labels ? where i can get the list

#6
by TahirC - opened

I am trying to find out which labels are supported

also is there a fine-tuning script ?

I got the same questions : is there a fine-tuning script ?

GLiNER doesn’t have a fixed, predefined set of labels. It’s prompt-driven: you can pass in any entity labels (types) you want at inference time (written in natural language), and the model predicts spans for those labels.

You can see this from the paper’s “Method / Model” description where the input format is defined as a list of entity types + the text (entity types are part of the prompt), and from the training data description (Pile-NER) which notes the dataset contains ~13k distinct entity types, showing the system is designed for open-ended labels rather than a closed schema.
paper: https://arxiv.org/pdf/2311.08526

You can try the fine tuning script and other example notebooks at https://github.com/urchade/GLiNER/blob/main/examples/finetune.ipynb

Sign up or log in to comment