--- library_name: onnx tags: - distilbert - ner - named-entity-recognition - token-classification - conll2003 - onnx - inference4j license: apache-2.0 pipeline_tag: token-classification --- # DistilBERT NER — ONNX ONNX export of [dslim/distilbert-NER](https://huggingface.co/dslim/distilbert-NER), a DistilBERT model fine-tuned on CoNLL-2003 for Named Entity Recognition. Identifies persons, organizations, locations, and miscellaneous entities in text using IOB2 tagging. Mirrored for use with [inference4j](https://github.com/inference4j/inference4j), an inference-only AI library for Java. ## Original Source - **Repository:** [dslim (ONNX by onnx-community)](https://huggingface.co/dslim/distilbert-NER) - **License:** apache-2.0 ## Usage with inference4j ```java try (BertNerRecognizer ner = BertNerRecognizer.builder() .modelId("inference4j/distilbert-NER") .build()) { List entities = ner.recognize("John works at Google in London."); for (NamedEntity e : entities) { System.out.printf("%s (%s)%n", e.text(), e.label()); } // John (PER) // Google (ORG) // London (LOC) } ``` ## Model Details | Property | Value | |----------|-------| | Architecture | DistilBERT (6 layers, 768 hidden, 66M params) | | Task | Named Entity Recognition (IOB2 tagging) | | Labels | O, B-PER, I-PER, B-ORG, I-ORG, B-LOC, I-LOC, B-MISC, I-MISC | | Training data | CoNLL-2003 | | F1 score | 92.17 | | Max sequence length | 512 | | Tokenizer | WordPiece (cased) | | Original framework | PyTorch (HuggingFace Transformers) | ## License This model is licensed under the [Apache License 2.0](https://www.apache.org/licenses/LICENSE-2.0). Original model by [dslim](https://huggingface.co/dslim/distilbert-NER), ONNX export by [onnx-community](https://huggingface.co/onnx-community).