使用BinaryRelevance包时遭遇ValueError(类别数需大于1)的解决求助
BinaryRelevance包时遭遇ValueError(类别数需大于1)的解决求助
我最近在跟着一篇多标签文本分类的教程,用自己的电影数据集做实验。把数据二值化之后,我尝试运行BinaryRelevance包,结果遇到了这个错误:The number of classes has to be greater than one; got 1 class
我已经试了下面这些解决办法,但问题还是没解决:
- 移除只有一个样本的类别:原本有34个标签,处理后剩下32个。我不仅删掉了对应这两个类别的列,还把 genres 列从分隔字符串拆分成单个类别(也就是展开行),然后删掉了包含这些稀有类别的行。
- 因为已经把genres列展开了,我尝试用分层的训练测试拆分方法来划分数据集,代码大概是这样的:
train, test = train_test_split(movies, random_state=42, ...
有没有大佬能帮我看看哪里还有问题呀?
备注:内容来源于stack exchange,提问作者minty0394
相关产品推荐
相关产品推荐

