explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-11-12 04:38:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

Do not add the root token to the adjacency map

Browse Source
This commit is contained in:
Raphaël Bournhonesque 2017-06-11 18:30:01 +02:00
parent d9c567371f
commit 4ca8a396a2
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
spacy/pattern/pattern.py
View File

@ -104,8 +104,10 @@ class DependencyTree(Tree):
for token in doc: for token in doc:
self.nodes[token.i] = token self.nodes[token.i] = token
# inverse the dependency to have an actual tree
self.adjacency[token.head.i][token.i] = token.dep_ if token.head.i != token.i:
# inverse the dependency to have an actual tree
self.adjacency[token.head.i][token.i] = token.dep_
def __getitem__(self, item): def __getitem__(self, item):
return self.nodes[item] return self.nodes[item]