ner-swedish-wikiann

ner-swedish-wikiann

This model is a fine-tuned version of nordic-roberta-wiki trained for NER on the wikiann dataset.

eval F1-Score: 83,78

test F1-Score: 83,76

Model Usage

from transformers import AutoTokenizer, AutoModelForTokenClassification

from transformers import pipeline

tokenizer = AutoTokenizer.from_pretrained("birgermoell/ner-swedish-wikiann")

model = AutoModelForTokenClassification.from_pretrained("birgermoell/ner-swedish-wikiann")

nlp = pipeline("ner", model=model, tokenizer=tokenizer)

example = "Jag heter Per och jag jobbar på KTH"

nlp(example)