Commit
•
546fb92
1
Parent(s):
63ae242
Update model card
Browse files
README.md
CHANGED
@@ -1,9 +1,30 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
Fine-tuning bert-base-multilingual-cased on Wikiann dataset for performing NER on Marathi language.
|
2 |
|
|
|
|
|
|
|
3 |
Label list: (0), B-PER (1), I-PER (2), B-ORG (3), I-ORG (4), B-LOC (5), I-LOC (6)
|
4 |
|
5 |
-
|
|
|
|
|
6 |
from transformers import AutoTokenizer, AutoModelForTokenClassification
|
|
|
7 |
|
8 |
tokenizer = AutoTokenizer.from_pretrained("lakshaywadhwa1993/ner_marathi_bert")
|
9 |
|
@@ -11,8 +32,9 @@ model = AutoModelForTokenClassification.from_pretrained("lakshaywadhwa1993/ner_m
|
|
11 |
|
12 |
nlp = pipeline("ner", model=model, tokenizer=tokenizer)
|
13 |
|
14 |
-
example = ["राज्यसभा","निवडणुकांसाठी","मुंबईत","भाजपचे" ,"चिंचवडचे", "आमदार", "लक्ष्मण", "जगताप"]
|
15 |
|
16 |
results = nlp(example)
|
17 |
|
18 |
-
results
|
|
|
|
1 |
+
---
|
2 |
+
language: mr
|
3 |
+
datasets:
|
4 |
+
- wikiann
|
5 |
+
examples:
|
6 |
+
widget:
|
7 |
+
- text: "सचिन तेंडुलकर मुंबईचा आहे."
|
8 |
+
example_title: "Sentence_1"
|
9 |
+
- text: "विराट कोहली भारताकडून खेळतो."
|
10 |
+
example_title: "Sentence_2"
|
11 |
+
- text: "नवी दिल्ली ही भारताची राजधानी आहे"
|
12 |
+
example_title: "Sentence_3"
|
13 |
+
---
|
14 |
+
|
15 |
+
<h1>Marathi Named Entity Recognition Model trained using transfer learning</h1>
|
16 |
Fine-tuning bert-base-multilingual-cased on Wikiann dataset for performing NER on Marathi language.
|
17 |
|
18 |
+
|
19 |
+
## Label ID and its corresponding label name
|
20 |
+
|
21 |
Label list: (0), B-PER (1), I-PER (2), B-ORG (3), I-ORG (4), B-LOC (5), I-LOC (6)
|
22 |
|
23 |
+
|
24 |
+
Example
|
25 |
+
```py
|
26 |
from transformers import AutoTokenizer, AutoModelForTokenClassification
|
27 |
+
from transformers import pipeline
|
28 |
|
29 |
tokenizer = AutoTokenizer.from_pretrained("lakshaywadhwa1993/ner_marathi_bert")
|
30 |
|
|
|
32 |
|
33 |
nlp = pipeline("ner", model=model, tokenizer=tokenizer)
|
34 |
|
35 |
+
example = ["राज्यसभा","निवडणुकांसाठी","मुंबईत","भाजपचे" ,"चिंचवडचे", "आमदार", "लक्ष्मण", "जगताप"]
|
36 |
|
37 |
results = nlp(example)
|
38 |
|
39 |
+
results
|
40 |
+
```
|