Chinese-Text-Classification-Pytorch icon indicating copy to clipboard operation
Chinese-Text-Classification-Pytorch copied to clipboard

Published 20 hours ago • verified publisher icon 649453932

一个文本要是有多个标签要怎么用这个模型呢

Open Deerkangkang opened this issue 1 year ago • 3 comments

如题，今天在格式化数据的时候发现，该仓库的数据都是一条文本对应一个类别， 如果一个本文对应多个类别时候怎么办

Deerkangkang avatar Mar 13 '23 06:03 Deerkangkang

同问，请问大佬解决了吗

Eminem-lamar avatar May 16 '23 13:05 Eminem-lamar

解决了，不过得自己写一个函数，比如说把输出的output = model（input） 的向量卡一个阈值设置为1，其余为0，然后标签使用的sklearn库的multilabel 什么的函数，可以自动把一个列表值变成和输出同维度的one-hot，然后就可以计算了

Deerkangkang avatar May 19 '23 01:05 Deerkangkang

啊好的谢谢大佬，可以看看您是怎么写的吗，因为我是第一次接触这个，没有什么经验

Eminem-lamar avatar May 23 '23 07:05 Eminem-lamar