README.md 1.07 KB
Newer Older
1
2
3
4
---
language:
- hi
- en
5
datasets:
6
- lince
Julien Chaumond's avatar
Julien Chaumond committed
7
license: mit
8
9
10
11
tags:
- codeswitching
- hindi-english
- ner
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
---

# codeswitch-hineng-ner-lince
This is a pretrained model for **Name Entity Recognition** of `Hindi-english` code-mixed data used from [LinCE](https://ritual.uh.edu/lince/home)

This model is trained for this below repository. 

[https://github.com/sagorbrur/codeswitch](https://github.com/sagorbrur/codeswitch)

To install codeswitch:

```
pip install codeswitch
```

27
## Name Entity Recognition of Code-Mixed Data
28

29
* **Method-1**
30
31
32

```py

33
from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline
34
35
36
37
38
39
40
41
42
43
44

tokenizer = AutoTokenizer.from_pretrained("sagorsarker/codeswitch-hineng-ner-lince")

model = AutoModelForTokenClassification.from_pretrained("sagorsarker/codeswitch-hineng-ner-lince")

ner_model = pipeline('ner', model=model, tokenizer=tokenizer)

ner_model("put any hindi english code-mixed sentence")

```

45
* **Method-2**
46
47
48
49
50
51
52
53
54

```py
from codeswitch.codeswitch import NER
ner = NER('hin-eng')
text = "" # your mixed sentence 
result = ner.tag(text)
print(result)

```