Closed gugudexiatian closed 5 years ago
demo: 以text-cnn为例写了个小demo,代码在/test/multi-class/目录下,将多标签转化为multi-hot类标,然后使用sigmoid、top_k_categorical_accuracy等作多类分类,阀值凭感觉取了0.5。
资料: 有时间的话可以试试,你可以看看sklearn机器学习实现scikit-multilearn,和sklearn-hierarchical-classification,详细讲解可以看下这个解决多标签分类问题(包括案例研究)。或者是腾讯不久前出的pytorch版NeuralNLP-NeuralClassifier。
多标签分类算法方案:
非常感谢您的及时答疑和提供的思路,受益匪浅,感谢感谢!!
感觉这样成体系的代码好棒,如果想改成多标签该如何修改代码呢