telugu_bertu_pos / README.md
julien-c's picture
julien-c HF staff
Migrate model card from transformers-repo
59dc25e
|
raw
history blame
1.12 kB

Part of Speech tagging Model for Telugu

How to use

from simpletransformers.ner import NERModel
model = NERModel('bert',
                 'kuppuluri/telugu_bertu_pos',
                 args={"use_multiprocessing": False},
                 labels=[
                     'QC', 'JJ', 'NN', 'QF', 'RDP', 'O',
                     'NNO', 'PRP', 'RP', 'VM', 'WQ',
                     'PSP', 'UT', 'CC', 'INTF', 'SYMP',
                     'NNP', 'INJ', 'SYM', 'CL', 'QO',
                     'DEM', 'RB', 'NST', ],
                 use_cuda=False)

text = "విరాట్ కోహ్లీ కూడా అదే నిర్లక్ష్యాన్ని ప్రదర్శించి కేవలం ఒక పరుగుకే రనౌటై పెవిలియన్ చేరాడు ."
results = model.predict([text])

Training data

Training data is from https://github.com/anikethjr/NER_Telugu

Eval results

On the test set my results were

eval_loss = 0.0036797842364565416

f1_score = 0.9983795127912227

precision = 0.9984325602401637

recall = 0.9983264709788816