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

* Respect the model_dir input parameter to train.py

Browse Source
This commit is contained in:
Matthew Honnibal 2015-04-08 22:48:26 +02:00
parent ed8942a096
commit a6ac92f077
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
bin/parser/train.py
View File

@ -212,7 +212,7 @@ def train(Language, train_loc, model_dir, n_iter=15, feat_set=u'basic', seed=0,
if n_sents > 0: if n_sents > 0:
gold_tuples = gold_tuples[:n_sents] gold_tuples = gold_tuples[:n_sents]
nlp = Language() nlp = Language(data_dir=model_dir)
print "Itn.\tUAS\tNER F.\tTag %" print "Itn.\tUAS\tNER F.\tTag %"
for itn in range(n_iter): for itn in range(n_iter):
@ -244,7 +244,7 @@ def train(Language, train_loc, model_dir, n_iter=15, feat_set=u'basic', seed=0,
def evaluate(Language, dev_loc, model_dir, gold_preproc=False, verbose=True): def evaluate(Language, dev_loc, model_dir, gold_preproc=False, verbose=True):
assert not gold_preproc assert not gold_preproc
nlp = Language() nlp = Language(data_dir=model_dir)
gold_tuples = read_docparse_file(dev_loc) gold_tuples = read_docparse_file(dev_loc)
scorer = Scorer() scorer = Scorer()
for raw_text, segmented_text, annot_tuples in gold_tuples: for raw_text, segmented_text, annot_tuples in gold_tuples: