RoBERTa Base Ontonotes NER Pipeline

Description

This pretrained pipeline is built on the top of roberta_base_token_classifier_ontonotes model.

Predicted Entities

Download Copy S3 URI

How to use


pipeline = PretrainedPipeline("roberta_base_token_classifier_ontonotes_pipeline", lang = "en")

pipeline.annotate("My name is John and I have been working at John Snow Labs since November 2020.")

val pipeline = new PretrainedPipeline("roberta_base_token_classifier_ontonotes_pipeline", lang = "en")

pipeline.annotate("My name is John and I have been working at John Snow Labs since November 2020."))

Results


+--------------+---------+
|chunk         |ner_label|
+--------------+---------+
|John          |PERSON   |
|John Snow Labs|ORG      |
|November 2020 |DATE     |
+--------------+---------+

Model Information

Model Name: roberta_base_token_classifier_ontonotes_pipeline
Type: pipeline
Compatibility: Spark NLP 4.0.0+
License: Open Source
Edition: Official
Language: en
Size: 456.5 MB

Included Models

  • DocumentAssembler
  • SentenceDetectorDLModel
  • TokenizerModel
  • RoBertaForTokenClassification
  • NerConverter
  • Finisher