Fix default labels for parser and NER

This commit is contained in:
Matthew Honnibal 2016-10-12 19:12:40 +02:00
parent 3a03c668c3
commit 0e2bedc373
1 changed files with 11 additions and 2 deletions

View File

@ -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')