English NER (7 classes) β€” Flukes

Named entity recognition for English text. Predicts seven entity types:

Label Description
PER Person names
ORG Organization names
LOC Locations (cities, countries, geographical features)
EVENT Named events (elections, wars, conferences, ...)
PRODUCT Named products
WORK_OF_ART Books, films, songs, paintings, ...
MISC All other entities (nationalities, languages, etc.)

⚠️ Default license: noncommercial use only. This model is released under the Flukes NC 1.0 License. Commercial use β€” including using this model's predictions in a commercial product or service β€” requires a separate license. Contact alan.akbik@gmail.com.


Use

Install flukes:

pip install flukes

Then:

from flukes.models import Model

model = Model.load("ner")   
doc = model.predict("Washington bought himself a Chevy Roadster.\n\nHe drove it to the Berlin Marathon.")

print(doc)

Output:

Document[80]: Washington bought himself a Chevy Roadster.
              ╰───PER───╯                  ╰───PRODUCT──╯

              He drove it to the Berlin Marathon.
                                 ╰─────EVENT────╯

Access Annotations

You can also access all annotations directly:

from flukes.models import Model

model = Model.load("ner")
doc = model.predict("Washington bought himself a Chevy Roadster.\n\nHe drove it to the Berlin Marathon.")

for span in doc.spans:
    print(f"{span.text!r:30}  {span.label}  ({span.score:.2f})")

Output:

'Washington'                    PER  (1.00)
'Chevy Roadster'                PRODUCT  (1.00)
'Berlin Marathon'               EVENT  (1.00)

Performance

This model scores 95.44 F1 on our internal test sets.

License

Model weights: Flukes Noncommercial License 1.0. Personal, academic, and other noncommercial use permitted. Commercial use requires a separate license β€” contact alan.akbik@gmail.com.

Flukes library code: Apache 2.0.

Downloads last month
4
Safetensors
Model size
0.6B params
Tensor type
F32
Β·
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support