首页 PaddleHub 帖子详情
PaddleHub 1.8.0,文本分类报错
收藏
快速回复
PaddleHub 其他预训练模型 1400 11
PaddleHub 1.8.0,文本分类报错
收藏
快速回复
PaddleHub 其他预训练模型 1400 11

如下代码在PaddleHub 1.6.2上面是好使的,移植到PaddleHub 1.8.0后报错(AttributeError: 'list' object has no attribute 'argmax')

不知道是什么原因,请大佬指点一下。

run_states = cls_task.predict(data=data)

results = [run_state.run_results for run_state in run_states]

for result in results:

     index = 0

    result = np.argmax(result, axis=2)[0]

    for label in result:

        print("%s\tp=%s" % (data[index][0], label_map[label]))

        index += 1

0
收藏
回复
全部评论(11)
时间顺序
f
fisher030712
#2 回复于2020-11

不知道PaddleHub1.8.0中,TextClassifierTask返回的结果,是不是跟1.6.2中的不同造成的?

该如何处理呢?

0
回复
AIStudio810258
#3 回复于2020-11

1.6、1.8有比较大的区别,一些地方没有向下兼容

0
回复
AIStudio810258
#4 回复于2020-11

官方说2.0以后版本就比较完善的向下兼容了

0
回复
AIStudio810258
#5 回复于2020-11

原来版本是指的hub啊。。。

0
回复
AIStudio810258
#6 回复于2020-11

报错上看是返回类型的问题,期望返回numpy类型,结果返回了list类型

0
回复
f
fisher030712
#7 回复于2020-11

不用for循环, 把results中的结果单独取出来,就好使了。

0
回复
AIStudio810258
#8 回复于2020-11

是不是返回的数据形状有了变化,可以打印下形状看看

0
回复
thinc
#9 回复于2020-11

可以去读一下1.8版本的使用教程看看

0
回复
thinc
#10 回复于2020-11
不用for循环, 把results中的结果单独取出来,就好使了。

解决了就行

0
回复
七年期限
#11 回复于2020-11
不用for循环, 把results中的结果单独取出来,就好使了。

0
回复
screamdw
#12 回复于2020-11

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户