NuNER_Zero-span / README.md
Serega6678's picture
Update README.md
a939888 verified
metadata
license: mit
datasets:
  - numind/NuNER
library_name: gliner
language:
  - en
pipeline_tag: token-classification
tags:
  - entity recognition
  - NER
  - named entity recognition
  - zero shot
  - zero-shot

NuZero - is the family of Zero-Shot Entity Recognition models inspired by GLiNER and built with insights we gathered throughout our work on NuNER.

NuZero span is a more powerful version of GLiNER-large-v2.1, surpassing it by 4% on average, and is trained on the diverse internal dataset tailored for real-life use cases.

Installation & Usage

!pip install gliner
from gliner import GLiNER

model = GLiNER.from_pretrained("numind/NuZero_span")

text = """

"""

labels = ["person", "award", "date", "competitions", "teams"]

entities = model.predict_entities(text, labels)

for entity in entities:
    print(entity["text"], "=>", entity["label"])

Fine-tuning

Citation

@misc{bogdanov2024nuner,
      title={NuNER: Entity Recognition Encoder Pre-training via LLM-Annotated Data}, 
      author={Sergei Bogdanov and Alexandre Constantin and Timothée Bernard and Benoit Crabbé and Etienne Bernard},
      year={2024},
      eprint={2402.15343},
      archivePrefix={arXiv},
      primaryClass={cs.CL}
}