图像分类项目中train_test_split报错:样本数量不一致[3556,3000]
解决train_test_split样本数不匹配的ValueError错误
这个错误明确说明特征集X和标签集data_target的样本数量不一致——X有3556个样本,data_target只有3000个,sklearn的train_test_split要求特征与标签的样本数必须完全相等才能执行划分。
解决步骤:
- 先确认样本数差异:在划分前打印两者的样本数量,明确问题
print("X的样本数:", X.shape[0]) print("data_target的样本数:", len(data_target)) - 排查不匹配原因:
- 检查图像数据加载环节,是否存在部分图像加载失败但标签未同步剔除的情况
- 核对标签文件与图像文件的对应关系,确认没有漏标或多标
- 检查数据预处理流程,是否过滤了部分特征数据但未同步处理标签
- 修正样本数:根据排查结果,要么补全缺失的标签/图像,要么删除多余的特征/标签,确保两者样本数完全一致后再执行
train_test_split划分
内容的提问来源于stack exchange,提问作者Sumeet Yadav
相关产品推荐
相关产品推荐

