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

* Assign root words the ROOT label via the Break transition. Something is still wrong here...

This commit is contained in:
Matthew Honnibal 2015-03-20 01:15:27 +01:00
parent 52429625f0
commit 9f4ad8fdfb
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/syntax/arc_eager.pyx
View File

@ -43,7 +43,7 @@ cdef class ArcEager(TransitionSystem):
@classmethod @classmethod
def get_labels(cls, gold_parses): def get_labels(cls, gold_parses):
move_labels = {SHIFT: {'': True}, REDUCE: {'': True}, RIGHT: {}, move_labels = {SHIFT: {'': True}, REDUCE: {'': True}, RIGHT: {},
LEFT: {}, BREAK: {'': True}} LEFT: {}, BREAK: {'ROOT': True}}
for raw_text, segmented, (ids, tags, heads, labels, iob) in gold_parses: for raw_text, segmented, (ids, tags, heads, labels, iob) in gold_parses:
for i, (head, label) in enumerate(zip(heads, labels)): for i, (head, label) in enumerate(zip(heads, labels)):
if label != 'ROOT': if label != 'ROOT':
@ -138,7 +138,7 @@ cdef int _do_break(const Transition* self, State* state) except -1:
state.sent[state.i-1].sent_end = True state.sent[state.i-1].sent_end = True
while state.stack_len != 0: while state.stack_len != 0:
if get_s0(state).head == 0: if get_s0(state).head == 0:
get_s0(state).dep = 0 get_s0(state).dep = self.label
state.stack -= 1 state.stack -= 1
state.stack_len -= 1 state.stack_len -= 1
if not at_eol(state): if not at_eol(state):