Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

* Fix bug in ArcEager.get_labels

This commit is contained in:
Matthew Honnibal 2015-03-27 15:21:06 +01:00
parent cd054c6c9f
commit b32f581acb
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
spacy/syntax/arc_eager.pyx
View File

@ -45,11 +45,11 @@ cdef class ArcEager(TransitionSystem):
move_labels = {SHIFT: {'': True}, REDUCE: {'': True}, RIGHT: {},
LEFT: {}, BREAK: {'ROOT': True}}
for raw_text, segmented, (ids, words, tags, heads, labels, iob) in gold_parses:
for i, (head, label) in enumerate(zip(heads, labels)):
for child, head, label in zip(ids, heads, labels):
if label != 'ROOT':
if head > i:
if head < child:
move_labels[RIGHT][label] = True
elif head < i:
elif head > child:
move_labels[LEFT][label] = True
return move_labels