-
Notifications
You must be signed in to change notification settings - Fork 372
/
exc_03_07.py
31 lines (26 loc) · 1.05 KB
/
exc_03_07.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
import spacy
from spacy.language import Language
from spacy.matcher import PhraseMatcher
from spacy.tokens import Span
nlp = spacy.load("fr_core_news_sm")
animals = ["Golden Retriever", "chat", "tortue", "Rattus norvegicus"]
animal_patterns = list(nlp.pipe(animals))
print("animal_patterns :", animal_patterns)
matcher = PhraseMatcher(nlp.vocab)
matcher.add("ANIMAL", animal_patterns)
# Définis le composant personnalisé
@Language.component("animal_component")
def animal_component_function(doc):
# Applique le matcher au doc
matches = ____
# Crée un Span pour chaque correpondance et assigne-lui le label "ANIMAL"
spans = [Span(____, ____, ___, label=____) for match_id, start, end in matches]
# Actualise doc.ents avec les spans en correspondance
doc.ents = spans
return doc
# Ajoute le composant au pipeline après le composant "ner"
____.____(____, ____=____)
print(nlp.pipe_names)
# Traite le texte et affiche le texte et le label pour les doc.ents
doc = nlp("J'ai un chat et un Golden Retriever")
print([(____, ____) for ent in ____])