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

649453932 / Chinese-Text-Classification-Pytorch

中文文本分类，TextCNN，TextRNN，FastText，TextRCNN，BiLSTM_Attention，DPCNN，Transformer，基于pytorch，开箱即用。

MIT License

5.19k stars 1.22k forks source link

Open Deerkangkang opened 1 year ago

Deerkangkang commented 1 year ago

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

Eminem-lamar commented 1 year ago

同问，请问大佬解决了吗

Deerkangkang commented 1 year ago

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

Eminem-lamar commented 1 year ago

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