Description
Onto is a Named Entity Recognition (or NER) model trained on OntoNotes 5.0. It can extract up to 18 entities such as people, places, organizations, money, time, date, etc.
This model uses the pretrained bert_base_cased embeddings model from BertEmbeddings annotator as an input.
Predicted Entities
CARDINAL, DATE, EVENT, FAC, GPE, LANGUAGE, LAW, LOC, MONEY, NORP, ORDINAL, ORG, PERCENT, PERSON, PRODUCT, QUANTITY, TIME, WORK_OF_ART.
Live Demo Open in Colab Download Copy S3 URI
How to use
...
embeddings = BertEmbeddings.pretrained("bert_base_cased", "en") \
.setInputCols("sentence", "token") \
.setOutputCol("embeddings")
ner_onto = NerDLModel.pretrained("onto_bert_base_cased", "en") \
.setInputCols(["document", "token", "embeddings"]) \
.setOutputCol("ner")
...
nlp_pipeline = Pipeline(stages=[document_assembler, sentence_detector, tokenizer, embeddings, ner_onto, ner_converter])
pipeline_model = nlp_pipeline.fit(spark.createDataFrame([['']]).toDF('text'))
result = pipeline_model.transform(spark.createDataFrame([["William Henry Gates III (born October 28, 1955) is an American business magnate, software developer, investor, and philanthropist. He is best known as the co-founder of Microsoft Corporation. During his career at Microsoft, Gates held the positions of chairman, chief executive officer (CEO), president and chief software architect, while also being the largest individual shareholder until May 2014. He is one of the best-known entrepreneurs and pioneers of the microcomputer revolution of the 1970s and 1980s. Born and raised in Seattle, Washington, Gates co-founded Microsoft with childhood friend Paul Allen in 1975, in Albuquerque, New Mexico; it went on to become the world's largest personal computer software company. Gates led the company as chairman and CEO until stepping down as CEO in January 2000, but he remained chairman and became chief software architect. During the late 1990s, Gates had been criticized for his business tactics, which have been considered anti-competitive. This opinion has been upheld by numerous court rulings. In June 2006, Gates announced that he would be transitioning to a part-time role at Microsoft and full-time work at the Bill & Melinda Gates Foundation, the private charitable foundation that he and his wife, Melinda Gates, established in 2000. He gradually transferred his duties to Ray Ozzie and Craig Mundie. He stepped down as chairman of Microsoft in February 2014 and assumed a new post as technology adviser to support the newly appointed CEO Satya Nadella."]], ["text"]))
...
val embeddings = BertEmbeddings.pretrained("bert_base_cased", "en")
.setInputCols(Array("sentence", "token"))
.setOutputCol("embeddings")
val ner_onto = NerDLModel.pretrained("onto_bert_base_cased", "en")
.setInputCols(Array("document", "token", "embeddings"))
.setOutputCol("ner")
...
val pipeline = new Pipeline().setStages(Array(document_assembler, sentence_detector, tokenizer, embeddings, ner_onto, ner_converter))
val data = Seq("William Henry Gates III (born October 28, 1955) is an American business magnate, software developer, investor, and philanthropist. He is best known as the co-founder of Microsoft Corporation. During his career at Microsoft, Gates held the positions of chairman, chief executive officer (CEO), president and chief software architect, while also being the largest individual shareholder until May 2014. He is one of the best-known entrepreneurs and pioneers of the microcomputer revolution of the 1970s and 1980s. Born and raised in Seattle, Washington, Gates co-founded Microsoft with childhood friend Paul Allen in 1975, in Albuquerque, New Mexico; it went on to become the world's largest personal computer software company. Gates led the company as chairman and CEO until stepping down as CEO in January 2000, but he remained chairman and became chief software architect. During the late 1990s, Gates had been criticized for his business tactics, which have been considered anti-competitive. This opinion has been upheld by numerous court rulings. In June 2006, Gates announced that he would be transitioning to a part-time role at Microsoft and full-time work at the Bill & Melinda Gates Foundation, the private charitable foundation that he and his wife, Melinda Gates, established in 2000. He gradually transferred his duties to Ray Ozzie and Craig Mundie. He stepped down as chairman of Microsoft in February 2014 and assumed a new post as technology adviser to support the newly appointed CEO Satya Nadella.").toDF("text")
val result = pipeline.fit(data).transform(data)
import nlu
text = ["""William Henry Gates III (born October 28, 1955) is an American business magnate, software developer, investor, and philanthropist. He is best known as the co-founder of Microsoft Corporation. During his career at Microsoft, Gates held the positions of chairman, chief executive officer (CEO), president and chief software architect, while also being the largest individual shareholder until May 2014. He is one of the best-known entrepreneurs and pioneers of the microcomputer revolution of the 1970s and 1980s. Born and raised in Seattle, Washington, Gates co-founded Microsoft with childhood friend Paul Allen in 1975, in Albuquerque, New Mexico; it went on to become the world's largest personal computer software company. Gates led the company as chairman and CEO until stepping down as CEO in January 2000, but he remained chairman and became chief software architect. During the late 1990s, Gates had been criticized for his business tactics, which have been considered anti-competitive. This opinion has been upheld by numerous court rulings. In June 2006, Gates announced that he would be transitioning to a part-time role at Microsoft and full-time work at the Bill & Melinda Gates Foundation, the private charitable foundation that he and his wife, Melinda Gates, established in 2000. He gradually transferred his duties to Ray Ozzie and Craig Mundie. He stepped down as chairman of Microsoft in February 2014 and assumed a new post as technology adviser to support the newly appointed CEO Satya Nadella."""]
ner_df = nlu.load('en.ner.onto.bert.cased_base').predict(text, output_level='chunk')
ner_df[["entities", "entities_class"]]
Results
+-----------------------+---------+
|chunk |ner_label|
+-----------------------+---------+
|William Henry Gates III|PERSON |
|October 28, 1955 |DATE |
|American |NORP |
|Microsoft Corporation |ORG |
|Microsoft |ORG |
|Gates |PERSON |
|May 2014 |DATE |
|one |CARDINAL |
|the 1970s and 1980s |DATE |
|Seattle |GPE |
|Washington |GPE |
|Gates |PERSON |
|Paul Allen |PERSON |
|1975 |DATE |
|Albuquerque |GPE |
|New Mexico |GPE |
|Gates |ORG |
|January 2000 |DATE |
|the late 1990s |DATE |
|Gates |PERSON |
+-----------------------+---------+
Model Information
| Model Name: | onto_bert_base_cased |
| Type: | ner |
| Compatibility: | Spark NLP 2.7.0+ |
| License: | Open Source |
| Edition: | Official |
| Input Labels: | [sentence, token, embeddings] |
| Output Labels: | [ner] |
| Language: | en |
Data Source
The model is trained based on data from OntoNotes 5.0
Benchmarking
Micro-average:
prec: 0.8987879, rec: 0.90063596, f1: 0.89971095
CoNLL Eval:
processed 152728 tokens with 11257 phrases; found: 11276 phrases; correct: 10006.
accuracy: 98.01%; 10006 11257 11276 precision: 88.74%; recall: 88.89%; FB1: 88.81
CARDINAL: 822 935 990 precision: 83.03%; recall: 87.91%; FB1: 85.40 990
DATE: 1355 1602 1567 precision: 86.47%; recall: 84.58%; FB1: 85.52 1567
EVENT: 32 63 59 precision: 54.24%; recall: 50.79%; FB1: 52.46 59
FAC: 96 135 124 precision: 77.42%; recall: 71.11%; FB1: 74.13 124
GPE: 2116 2240 2182 precision: 96.98%; recall: 94.46%; FB1: 95.70 2182
LANGUAGE: 10 22 11 precision: 90.91%; recall: 45.45%; FB1: 60.61 11
LAW: 21 40 28 precision: 75.00%; recall: 52.50%; FB1: 61.76 28
LOC: 141 179 178 precision: 79.21%; recall: 78.77%; FB1: 78.99 178
MONEY: 278 314 321 precision: 86.60%; recall: 88.54%; FB1: 87.56 321
NORP: 799 841 850 precision: 94.00%; recall: 95.01%; FB1: 94.50 850
ORDINAL: 177 195 217 precision: 81.57%; recall: 90.77%; FB1: 85.92 217
ORG: 1606 1795 1848 precision: 86.90%; recall: 89.47%; FB1: 88.17 1848
PERCENT: 306 349 344 precision: 88.95%; recall: 87.68%; FB1: 88.31 344
PERSON: 1856 1988 1978 precision: 93.83%; recall: 93.36%; FB1: 93.60 1978
PRODUCT: 54 76 76 precision: 71.05%; recall: 71.05%; FB1: 71.05 76
QUANTITY: 87 105 108 precision: 80.56%; recall: 82.86%; FB1: 81.69 108
TIME: 143 212 216 precision: 66.20%; recall: 67.45%; FB1: 66.82 216
WORK_OF_ART: 107 166 179 precision: 59.78%; recall: 64.46%; FB1: 62.03 179