explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-11-10 19:57:17 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fixed NER model loading bug

Browse Source
This commit is contained in:
Vincent Genty 2017-09-26 15:46:04 +02:00
parent 2f8d535f65
commit 259ed027af
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
spacy/syntax/parser.pyx
View File

@ -147,6 +147,9 @@ cdef class Parser:
# TODO: remove this shim when we don't have to support older data
if 'labels' in cfg and 'actions' not in cfg:
cfg['actions'] = cfg.pop('labels')
# Convert string keys to int
if cfg.get('actions'):
cfg['actions'] = {int(action_name): labels for action_name, labels in cfg['actions'].items()}
# TODO: remove this shim when we don't have to support older data
for action_name, labels in dict(cfg.get('actions', {})).items():
# We need this to be sorted