PaddleHub 1.8.0,文本分类报错
收藏
如下代码在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
收藏
请登录后评论
不知道PaddleHub1.8.0中,TextClassifierTask返回的结果,是不是跟1.6.2中的不同造成的?
该如何处理呢?
1.6、1.8有比较大的区别,一些地方没有向下兼容
官方说2.0以后版本就比较完善的向下兼容了
原来版本是指的hub啊。。。
报错上看是返回类型的问题,期望返回numpy类型,结果返回了list类型
不用for循环, 把results中的结果单独取出来,就好使了。
是不是返回的数据形状有了变化,可以打印下形状看看
可以去读一下1.8版本的使用教程看看
解决了就行
顶
顶