1import spacy
2
3nlp = spacy.load("my_custom_el_model")
4doc = nlp("Ada Lovelace was born in London")
5
6# document level
7ents = [(e.text, e.label_, e.kb_id_) for e in doc.ents]
8print(ents) # [('Ada Lovelace', 'PERSON', 'Q7259'), ('London', 'GPE', 'Q84')]
9
10# token level
11ent_ada_0 = [doc[0].text, doc[0].ent_type_, doc[0].ent_kb_id_]
12ent_ada_1 = [doc[1].text, doc[1].ent_type_, doc[1].ent_kb_id_]
13ent_london_5 = [doc[5].text, doc[5].ent_type_, doc[5].ent_kb_id_]
14print(ent_ada_0) # ['Ada', 'PERSON', 'Q7259']
15print(ent_ada_1) # ['Lovelace', 'PERSON', 'Q7259']
16print(ent_london_5) # ['London', 'GPE', 'Q84']
17