From c57bf6485d5b9d7e8d5c3186e3c6c6f0903b634e Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Sat, 20 Aug 2016 04:16:50 +0200 Subject: [PATCH] Print dev score before averaging --- bin/parser/conll_train.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/bin/parser/conll_train.py b/bin/parser/conll_train.py index e8039c47c..20cd01653 100755 --- a/bin/parser/conll_train.py +++ b/bin/parser/conll_train.py @@ -172,10 +172,14 @@ def train(Language, gold_tuples, model_dir, dev_loc, n_iter=15, feat_set=u'basic except KeyboardInterrupt: print("Saving model...") break + dev_uas = score_file(nlp, dev_loc).uas + print("Dev before average", dev_uas) + nlp.parser.model.end_training() print("Saved. Evaluating...") return nlp + def _train_epoch(nlp, gold_tuples, eg_seen, itn, dev_loc, micro_eval): random.shuffle(gold_tuples) loss = 0