mirror of
https://github.com/explosion/spaCy.git
synced 2025-01-06 07:16:29 +03:00
40 lines
885 B
Python
40 lines
885 B
Python
import spacy
|
|
from spacy.kb import KnowledgeBase
|
|
|
|
|
|
def create_kb():
|
|
mykb = KnowledgeBase()
|
|
print("kb size", len(mykb))
|
|
|
|
entity_id = "Q42"
|
|
mykb.add_entity(entity_id=entity_id, prob=0.5)
|
|
print("adding entity", entity_id)
|
|
|
|
print("kb size", len(mykb))
|
|
|
|
|
|
def add_el():
|
|
nlp = spacy.load('en_core_web_sm')
|
|
print("pipes before:", nlp.pipe_names)
|
|
|
|
el_pipe = nlp.create_pipe(name='el')
|
|
nlp.add_pipe(el_pipe, last=True)
|
|
|
|
print("pipes after:", nlp.pipe_names)
|
|
print()
|
|
|
|
text = "The Hitchhiker's Guide to the Galaxy, written by Douglas Adams, reminds us to always bring our towel."
|
|
doc = nlp(text)
|
|
|
|
for token in doc:
|
|
print("token", token.text, token.ent_type_, token.ent_kb_id_)
|
|
|
|
print()
|
|
for ent in doc.ents:
|
|
print("ent", ent.text, ent.label_, ent.kb_id_)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# add_el()
|
|
create_kb()
|