BERT Base NER — ONNX

ONNX export of dslim/bert-base-NER, a BERT 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, an inference-only AI library for Java.

Original Source

• Repository: dslim (ONNX by Xenova)
• License: mit

Usage with inference4j

try (BertNerRecognizer ner = BertNerRecognizer.builder()
        .modelId("inference4j/bert-base-NER")
        .build()) {
    List<NamedEntity> entities = ner.recognize("John works at Google in London.");
    for (NamedEntity e : entities) {
        System.out.printf("%s (%s)%n", e.text(), e.label());
    }
}

Model Details

License

This model is licensed under the MIT License. Original model by dslim, ONNX export by Xenova.