提交 67c31c40 编写于 作者: Z Zeyu Chen

fix sequence labling task evaluation error

上级 1cc79ed0
...@@ -20,6 +20,8 @@ import time ...@@ -20,6 +20,8 @@ import time
import paddle.fluid as fluid import paddle.fluid as fluid
import paddlehub as hub import paddlehub as hub
import numpy as np
from paddlehub.common.logger import logger from paddlehub.common.logger import logger
......
...@@ -124,13 +124,13 @@ def _finetune_seq_label_task(task, ...@@ -124,13 +124,13 @@ def _finetune_seq_label_task(task,
exe=exe) exe=exe)
if do_eval and global_step % config.eval_interval == 0: if do_eval and global_step % config.eval_interval == 0:
evaluate_seq_label_task( evaluate_seq_labeling_task(
task, task,
data_reader, data_reader,
feed_list, feed_list,
phase="test", phase="test",
config=config) config=config)
evaluate_seq_label_task( evaluate_seq_labeling_task(
task, task,
data_reader, data_reader,
feed_list, feed_list,
...@@ -146,9 +146,9 @@ def _finetune_seq_label_task(task, ...@@ -146,9 +146,9 @@ def _finetune_seq_label_task(task,
exe=exe) exe=exe)
if do_eval: if do_eval:
evaluate_seq_label_task( evaluate_seq_labeling_task(
task, data_reader, feed_list, phase="dev", config=config) 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) task, data_reader, feed_list, phase="test", config=config)
logger.info("PaddleHub finetune finished.") logger.info("PaddleHub finetune finished.")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册