Hugging Face's logo

protectai
/
bert-large-NER-onnx

Token Classification
Transformers
ONNX
English
bert
ner
Eval Results
Model card Files Community
bert-large-NER-onnx
File size: 2,606 Bytes 
0a8190a
 
 
 
 
 
 
 
 
 
 
7882249
0a8190a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1a692cc
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
 
---
language:
- en
inference: false
pipeline_tag: token-classification
tags:
- ner
- bert
license: mit
datasets:
- conll2003
base_model: dslim/bert-large-NER
model-index:
- name: dslim/bert-large-NER
  results:
  - task:
      type: token-classification
      name: Token Classification
    dataset:
      name: conll2003
      type: conll2003
      config: conll2003
      split: test
    metrics:
    - name: Accuracy
      type: accuracy
      value: 0.9031688753722759
      verified: true
    - name: Precision
      type: precision
      value: 0.920025068328604
      verified: true
    - name: Recall
      type: recall
      value: 0.9193688678588825
      verified: true
    - name: F1
      type: f1
      value: 0.9196968510445761
      verified: true
    - name: loss
      type: loss
      value: 0.5085050463676453
      verified: true
---

# ONNX version of dslim/bert-large-NER

**This model is a conversion of [dslim/bert-large-NER](https://huggingface.co/dslim/bert-large-NER) to ONNX** format using the [🤗 Optimum](https://huggingface.co/docs/optimum/index) library.

**bert-large-NER** is a fine-tuned BERT model that is ready to use for **Named Entity Recognition** and achieves **state-of-the-art performance** for the NER task. It has been trained to recognize four types of entities: location (LOC), organizations (ORG), person (PER) and Miscellaneous (MISC). 

Specifically, this model is a *bert-large-cased* model that was fine-tuned on the English version of the standard [CoNLL-2003 Named Entity Recognition](https://www.aclweb.org/anthology/W03-0419.pdf) dataset. 

## Usage

Loading the model requires the [🤗 Optimum](https://huggingface.co/docs/optimum/index) library installed.

```python
from optimum.onnxruntime import ORTModelForTokenClassification
from transformers import AutoTokenizer, pipeline


tokenizer = AutoTokenizer.from_pretrained("laiyer/bert-large-NER-onnx")
model = ORTModelForTokenClassification.from_pretrained("laiyer/bert-large-NER-onnx")
ner = pipeline(
    task="ner",
    model=model,
    tokenizer=tokenizer,
)

ner_output = ner("My name is John Doe.")
print(ner_output)
```

### LLM Guard

[Anonymize scanner](https://llm-guard.com/input_scanners/anonymize/)

## Community

Join our Slack to give us feedback, connect with the maintainers and fellow users, ask questions, 
or engage in discussions about LLM security!

<a href="https://join.slack.com/t/laiyerai/shared_invite/zt-28jv3ci39-sVxXrLs3rQdaN3mIl9IT~w"><img src="https://github.com/laiyer-ai/llm-guard/blob/main/docs/assets/join-our-slack-community.png?raw=true" width="200"></a>