diff options
Diffstat (limited to 'stanza/models/ner_tagger.py')
-rw-r--r-- | stanza/models/ner_tagger.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/stanza/models/ner_tagger.py b/stanza/models/ner_tagger.py index 020e2c68..dd24dbbb 100644 --- a/stanza/models/ner_tagger.py +++ b/stanza/models/ner_tagger.py @@ -249,8 +249,9 @@ def train(args): logger.info("Training ended with {} steps.".format(global_step)) - best_f, best_eval = max(dev_score_history)*100, np.argmax(dev_score_history)+1 - logger.info("Best dev F1 = {:.2f}, at iteration = {}".format(best_f, best_eval * args['eval_interval'])) + if len(dev_score_history) > 0: + best_f, best_eval = max(dev_score_history)*100, np.argmax(dev_score_history)+1 + logger.info("Best dev F1 = {:.2f}, at iteration = {}".format(best_f, best_eval * args['eval_interval'])) def evaluate(args): # file paths |