You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

图像分类项目中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

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.07.17 03:14:50