From 67c31c4071b2769fe1aa02c48931db791f53f502 Mon Sep 17 00:00:00 2001 From: Zeyu Chen Date: Tue, 9 Apr 2019 14:48:48 +0800 Subject: [PATCH] fix sequence labling task evaluation error --- paddlehub/finetune/evaluate.py | 2 ++ paddlehub/finetune/finetune.py | 8 ++++---- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/paddlehub/finetune/evaluate.py b/paddlehub/finetune/evaluate.py index 448a127e..758a0b66 100644 --- a/paddlehub/finetune/evaluate.py +++ b/paddlehub/finetune/evaluate.py @@ -20,6 +20,8 @@ import time import paddle.fluid as fluid import paddlehub as hub +import numpy as np + from paddlehub.common.logger import logger diff --git a/paddlehub/finetune/finetune.py b/paddlehub/finetune/finetune.py index 114f6d46..367cbf90 100644 --- a/paddlehub/finetune/finetune.py +++ b/paddlehub/finetune/finetune.py @@ -124,13 +124,13 @@ def _finetune_seq_label_task(task, exe=exe) if do_eval and global_step % config.eval_interval == 0: - evaluate_seq_label_task( + evaluate_seq_labeling_task( task, data_reader, feed_list, phase="test", config=config) - evaluate_seq_label_task( + evaluate_seq_labeling_task( task, data_reader, feed_list, @@ -146,9 +146,9 @@ def _finetune_seq_label_task(task, exe=exe) if do_eval: - evaluate_seq_label_task( + evaluate_seq_labeling_task( task, data_reader, feed_list, phase="dev", config=config) - evaluate_seq_label_task( + evaluate_seq_labeling_task( task, data_reader, feed_list, phase="test", config=config) logger.info("PaddleHub finetune finished.") -- GitLab