vbius01 commited on
Commit
7d76965
·
verified ·
1 Parent(s): 1c18a20

Create README.md

Browse files

**Model description**
est-roberta-ud-ner is an [Est-RoBERTa](https://huggingface.co/EMBEDDIA/est-roberta) based model fine-tuned for named entity recognition on the [EDT](https://github.com/UniversalDependencies/UD_Estonian-EDT) and [EWT](https://github.com/UniversalDependencies/UD_Estonian-EWT) datasets.

**How to use**
You can use this model with Transformers pipeline for NER.
```
from transformers import pipeline

ner = pipeline("ner", model="vbius01/est-roberta-ud-ner")

text = "Eesti kuulub erinevalt Lätist ja Leedust kahtlemata Põhjamaade kultuuriruumi."
results = ner(text)

print(results)
```
```
[{'entity': 'B-GEP', 'score': np.float32(0.99339926), 'index': 1, 'word': '▁Eesti', 'start': 0, 'end': 5}, {'entity': 'B-GEP', 'score': np.float32(0.9923631), 'index': 4, 'word': '▁Lätist', 'start': 22, 'end': 29}, {'entity': 'B-GEP', 'score': np.float32(0.990756), 'index': 6, 'word': '▁Leedust', 'start': 32, 'end': 40}, {'entity': 'B-LOC', 'score': np.float32(0.61792), 'index': 8, 'word': '▁Põhjamaade', 'start': 51, 'end': 62}]
```

Files changed (1) hide show
  1. README.md +10 -0
README.md ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - et
4
+ base_model:
5
+ - EMBEDDIA/est-roberta
6
+ pipeline_tag: token-classification
7
+ library_name: transformers
8
+ tags:
9
+ - NER
10
+ ---