报错：Sklearn ValueError: This solver needs samples of at least 2 classes in the data, but the data

我是尘客哥

于 2020-10-01 17:57:39 发布

阅读量1.5k

点赞数

分类专栏：琐碎知识点记录文章标签： bug

本文链接：https://blog.csdn.net/icomma/article/details/108896293

版权

当使用Sklearn时，遇到ValueError提示数据集中至少需要2个类别的样本。这个问题通常由于数据集未经shuffle操作，导致在交叉验证过程中出现了单一类别样本。解决方案是先对数据集进行shuffle，以减少这种错误出现的概率。然而，即使shuffle，如果数据极度不均衡，该问题仍可能发生。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

错误原因

train_sizes, train_scores, test_scores = learning_curve(estimator, X, y, cv=cv, n_jobs=n_jobs, train_sizes=train_sizes, verbose=verbose) #请注意X,y

解决方法：

from sklearn

最低0.47元/天解锁文章