explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-26 18:06:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

* Set transition system early

Browse Source
This commit is contained in:
Matthew Honnibal 2015-06-15 02:54:12 +02:00
parent 38a6afa484
commit 4841f8ad5e
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
bin/parser/train.py
View File

@ -94,9 +94,6 @@ def train(Language, gold_tuples, model_dir, n_iter=15, feat_set=u'basic',
os.mkdir(pos_model_dir) os.mkdir(pos_model_dir)
os.mkdir(ner_model_dir) os.mkdir(ner_model_dir)
if use_orig_arc_eager:
Language.ParserTransitionSystem = OrigArcEager
setup_model_dir(sorted(POS_TAGS.keys()), POS_TAGS, POS_TEMPLATES, pos_model_dir) setup_model_dir(sorted(POS_TAGS.keys()), POS_TAGS, POS_TEMPLATES, pos_model_dir)
Config.write(dep_model_dir, 'config', features=feat_set, seed=seed, Config.write(dep_model_dir, 'config', features=feat_set, seed=seed,
@ -108,6 +105,8 @@ def train(Language, gold_tuples, model_dir, n_iter=15, feat_set=u'basic',
if n_sents > 0: if n_sents > 0:
gold_tuples = gold_tuples[:n_sents] gold_tuples = gold_tuples[:n_sents]
if use_orig_arc_eager:
Language.ParserTransitionSystem = OrigArcEager
nlp = Language(data_dir=model_dir) nlp = Language(data_dir=model_dir)