1
1
mirror of https://github.com/leon-ai/leon.git synced 2024-12-28 11:13:14 +03:00
leon/bridges/python/ner/spacy-run.py

13 lines
500 B
Python
Raw Normal View History

#!/usr/bin/env python
# -*- coding:utf-8 -*-
import spacy
nlp = spacy.load("en_core_web_trf", disable=["tagger", "parser", "attribute_ruler", "lemmatizer"])
# nlp = spacy.load("en_core_web_md", disable=["tok2vec", "tagger", "parser", "attribute_ruler", "lemmatizer"])
# doc = nlp("Do you know Spotify? I live in Paris and I'm Matthieu")
doc = nlp("Hi, I'm louis and I'm from saint-claude. Today I live in Shenzhen China and I work at Alibaba")
for ent in doc.ents:
print(ent.text, ent.label_)