首页 炼丹房 帖子详情
多标签分类,如何提取评估结果?
收藏
快速回复
炼丹房 问答新手上路 1755 6
多标签分类,如何提取评估结果?
收藏
快速回复
炼丹房 问答新手上路 1755 6

采用hub.MultiLabelClassifierTask进行多标签文本分类算法,提取的评估结果如下:

新款低配比以前低配配置方面高了好多,性能不错。 predict=[1, 2, 0, 2, 3, 0]

预测的结果不应该是0/1吗?怎么会出现2和3呢,感觉不太合理呀。

 

代码如下:

index = 0
run_states = multi_label_cls_task.predict(data=data)
results = [run_state.run_results for run_state in run_states]
for result in results:
    print("%s" % (result))
    # get predict index
    label_ids = []
    for i in range(dataset.num_labels):
        label_val = np.argmax(result[i])
        label_ids.append(label_val)
    
    print("%s\tpredict=%s" % (data[index][0], label_ids))
    index += 1

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

有没有人也遇到同样的问题?

0
回复
thinc
#3 回复于2020-08

多标签难道不是有0123吗。

0
回复
f
fisher030712
#4 回复于2020-08

那个输出结果是一个数组,数组元素的位置,就是对应标签的位置,数组中的值应该是0/1,出现2/3,就不太对了。

感觉应该是最后取结果的时候,用错了函数。我这个例子,是在网上找的。

0
回复
f
fisher030712
#5 回复于2020-08

感觉社区里面,对多标签分类,感兴趣的人不多啊。例子也没有。

0
回复
thinc
#6 回复于2020-08
感觉社区里面,对多标签分类,感兴趣的人不多啊。例子也没有。

去 AI Studio转转

0
回复
thinc
#7 回复于2020-08
那个输出结果是一个数组,数组元素的位置,就是对应标签的位置,数组中的值应该是0/1,出现2/3,就不太对了。 感觉应该是最后取结果的时候,用错了函数。我这个例子,是在网上找的。

是不是数据集出错了?在for循环中的result你看下打印出来的值是否有问题

0
回复
在@后输入用户全名并按空格结束,可艾特全站任一用户