如何解决如何解决 ValueError:类的数量必须大于 1;有 1 节课
当我运行以下命令时:
from sklearn.svm import SVC
from sklearn.model_selection import train_test_split
x_train,x_test,y_train,y_test=train_test_split(x,y,test_size=0.2)
clf=SVC(kernel='rbf',probability=True)
clf.fit(x_train,y_train)
我收到了 ValueError:The number of classes has to be greater than one; got 1 class
我使用了代码:print(np.unique(y_train))
,它返回了“[0]”。
任何人都可以为我指出解决方案的正确方向吗?
解决方法
要么您的 y
列表不包含 1,要么 y
中的 1 太少以至于 y_train
最终可能不包含 1。您应该打印 y
,如果它包含 1,则需要更改拆分策略以确保所有类至少出现在 y_train
和 y_test
中一次
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。