explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-09-21 19:39:13 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix parser labels in pipeline

Browse Source
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)