From 4f9657b42b79af4c640b67e1d2740b9d901d5aba Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Sun, 23 Apr 2017 22:27:10 +0200 Subject: [PATCH] Fix reporting if no dev data with train --- spacy/cli/train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/spacy/cli/train.py b/spacy/cli/train.py index 8557019c6..ce4c77416 100644 --- a/spacy/cli/train.py +++ b/spacy/cli/train.py @@ -2,6 +2,7 @@ from __future__ import unicode_literals, division, print_function import json +from collections import defaultdict from ..util import ensure_path from ..scorer import Scorer @@ -62,7 +63,7 @@ def train_model(Language, train_data, dev_data, output_path, tagger_cfg, parser_ for itn, epoch in enumerate(trainer.epochs(n_iter, augment_data=None)): for doc, gold in epoch: trainer.update(doc, gold) - dev_scores = trainer.evaluate(dev_data).scores if dev_data else {} + dev_scores = trainer.evaluate(dev_data).scores if dev_data else defaultdict(float) print_progress(itn, trainer.nlp.parser.model.nr_weight, trainer.nlp.parser.model.nr_active_feat, **dev_scores)