Deep Learning model developed for Named Entity Recognition (NER) in Slovak. The
Gerulata/SlovakBERT based model is fine-tuned on webscraped Slovak news articles. The finished model supports the following IOB tagged entity categories:
PERSON,
ORGANIZATION,
LOCATION,
DATE,
TIME,
MONEY and
PERCENTAGE
1from transformers import pipeline
2
3ner_pipeline = pipeline(task='ner', model='Raychani1/slovakbert-ner-v2')
4input_sentence = 'Hoci podľa ostatných údajov NBS pre Bratislavský kraj je aktuálna priemerná cena nehnuteľností na úrovni 2 072 eur za štvorcový meter, ceny bytov v hlavnom meste sú podstatne vyššie.'
5classifications = ner_pipeline(input_sentence)
For a Visualization Example please refer to the following
Gist.
Best model results are reached in the 8th training epoch.