From 0e2bedc373f16e45c955c9cf30a1219e884a3769 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Wed, 12 Oct 2016 19:12:40 +0200 Subject: [PATCH] Fix default labels for parser and NER --- spacy/language.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/spacy/language.py b/spacy/language.py index d07959aae..47557ec17 100644 --- a/spacy/language.py +++ b/spacy/language.py @@ -145,9 +145,18 @@ class BaseDefaults(object): tokenizer_exceptions = {} - parser_labels = {0: {'ROOT': True}} + parser_labels = {0: {'': True}, 1: {'': True}, 2: {'ROOT': True, 'nmod': True}, + 3: {'ROOT': True, 'nmod': True}, 4: {'ROOT': True}} + - entity_labels = {0: {'PER': True, 'LOC': True, 'ORG': True, 'MISC': True}} + entity_labels = { + 0: {'': True}, + 1: {'PER': True, 'LOC': True, 'ORG': True, 'MISC': True}, + 2: {'PER': True, 'LOC': True, 'ORG': True, 'MISC': True}, + 3: {'PER': True, 'LOC': True, 'ORG': True, 'MISC': True}, + 4: {'PER': True, 'LOC': True, 'ORG': True, 'MISC': True}, + 5: {'': True} + } parser_features = get_templates('parser')