Skip to content

Commit 192225c

Browse files
kuppulurstas00
authored andcommitted
Create README.md (huggingface#8168)
* Create README.md * Update README.md
1 parent 3a7308c commit 192225c

File tree

1 file changed

+35
-0
lines changed
  • model_cards/kuppuluri/telugu_bertu_ner

1 file changed

+35
-0
lines changed
Lines changed: 35 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,35 @@
1+
# Named Entity Recognition Model for Telugu
2+
3+
#### How to use
4+
5+
```python
6+
from simpletransformers.ner import NERModel
7+
model = NERModel('bert',
8+
'kuppuluri/telugu_bertu_ner',
9+
labels=[
10+
'B-PERSON', 'I-ORG', 'B-ORG', 'I-LOC', 'B-MISC',
11+
'I-MISC', 'I-PERSON', 'B-LOC', 'O'
12+
],
13+
use_cuda=False,
14+
args={"use_multiprocessing": False})
15+
16+
text = "విరాట్ కోహ్లీ కూడా అదే నిర్లక్ష్యాన్ని ప్రదర్శించి కేవలం ఒక పరుగుకే రనౌటై పెవిలియన్ చేరాడు ."
17+
results = model.predict([text])
18+
```
19+
20+
## Training data
21+
22+
Training data is from https://github.com/anikethjr/NER_Telugu
23+
24+
## Eval results
25+
26+
On the test set my results were
27+
28+
eval_loss = 0.0004407190410447974
29+
30+
f1_score = 0.999519076627124
31+
32+
precision = 0.9994389677005691
33+
34+
recall = 0.9995991983967936
35+

0 commit comments

Comments
 (0)