我正在尝试使用OneVsRestClassifier对一组评论进行多标签分类。我的目标是为每条评论标记可能的主题列表。我的自定义分类器使用一个csv文件中手动整理好的单词及其对应标签列表来标记每条评论。我试图将词袋技术获得的结果与我的自定义分类器的结果结合起来,使用VotingClassifier。这里是我现有代码的一部分:
我的意图是使用通过几种启发式方法获得的手动整理单词列表,以改进仅应用词袋技术所获得的结果。目前,我正在努力寻找一种方法来知道在预测时正在分类的标签,因为OneVsRestClassifier为每个标签创建了CustomClassifier的副本。
回答:
我想您在寻找classes_
属性:http://scikit-learn.org/dev/modules/generated/sklearn.multiclass.OneVsRestClassifier.html#sklearn.multiclass.OneVsRestClassifier