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

Fix parser labels in pipeline

This commit is contained in:
Matthew Honnibal 2016-10-16 17:03:22 +02:00
parent 01b42c531f
commit 4bb73b1a93
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
spacy/pipeline.pyx
View File

@ -28,14 +28,14 @@ cdef class DependencyParser(Parser):
@classmethod
def blank(cls, Vocab vocab, **cfg):
if 'actions' not in cfg:
cfg['actions'] = {1: {'': True}, 2: {'': True}, 3: {}, 4: {},
5: {'ROOT': True}}
cfg['actions'] = {0: {'': True}, 1: {'': True}, 2: {}, 3: {},
4: {'ROOT': True}}
for label in cfg.get('left_labels', []):
cfg['actions'][3][label] = True
cfg['actions'][2][label] = True
for label in cfg.get('right_labels', []):
cfg['actions'][4][label] = True
cfg['actions'][3][label] = True
for label in cfg.get('break_labels', []):
cfg['actions'][5][label] = True
cfg['actions'][4][label] = True
return Parser.blank(vocab, ArcEager, **cfg)