多标签分类,如何提取评估结果?
收藏
采用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
收藏
请登录后评论
有没有人也遇到同样的问题?
多标签难道不是有0123吗。
那个输出结果是一个数组,数组元素的位置,就是对应标签的位置,数组中的值应该是0/1,出现2/3,就不太对了。
感觉应该是最后取结果的时候,用错了函数。我这个例子,是在网上找的。
感觉社区里面,对多标签分类,感兴趣的人不多啊。例子也没有。
去 AI Studio转转
是不是数据集出错了?在for循环中的result你看下打印出来的值是否有问题