欢迎您对PaddleHub提出建议,非常感谢您对PaddleHub的贡献!
目前PaddleHub已实现您提出的需求,请安装PaddleHub>=1.7.0, PaddlePaddle>=1.8.0快速体验。请您在定义task任务时增加metrics_choices=['f1’]选项,即可实现多分类F1 Score评估指标,示例如下:
task = hub.ImageClassifierTask(
data_reader=data_reader,
feed_list=feed_list,
feature=feature_map,
num_classes=dataset.num_labels,
metrics_choices=['f1'], #PaddleHub同时支持'f1'和'acc'评价标准,可使用metrics_choices=['f1', 'acc']快速实现
config=config)
发现一个新的问题,ImageClassifierTask设定f1作为metrics时,多分类任务会报错,
metrics_choices = ['f1']
错误信息:
[2020-08-07 11:13:35,971] [ INFO] - PaddleHub finetune start
---------------------------------------------------------------------------TypeError Traceback (most recent call last) in<