From 4ca8a396a2934ddef3b7c71bab932f6bbe649759 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rapha=C3=ABl=20Bournhonesque?= Date: Sun, 11 Jun 2017 18:30:01 +0200 Subject: [PATCH] Do not add the root token to the adjacency map --- spacy/pattern/pattern.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/spacy/pattern/pattern.py b/spacy/pattern/pattern.py index de7a54e05..f21edf5a6 100644 --- a/spacy/pattern/pattern.py +++ b/spacy/pattern/pattern.py @@ -104,8 +104,10 @@ class DependencyTree(Tree): for token in doc: 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): return self.nodes[item]