ValueError: With n_samples=0, test_size=0.2 and train_size=None, the resulting train set will be emp

今天写代码labelmetovoc,即将labelme标注的转化为voc标准格式参考的这篇文章时遇到了如下问题:
ValueError: With n_samples=0, test_size=0.2 and train_size=None, the resulting train set will be empty. Adjust any of the aforementioned parameters.
在网上查了一下,大部分博客都认为是scikit-learn版本较高时出现的问题,需要换到0.20.0以下版本,但是换的话numpy,scipy等库均要重新下载兼容的版本,所以建议不要轻易尝试,我请教了一下师兄,他说是因为参数上面参数设置出现了问题,train或test的数量必须为整数,改好的代码如下所示:

# -*- coding: utf-8 -*-
"""
Created on Fri Apr 10 13:39:17 2020

@author: nihao
"""
import os
import numpy as np
import codecs
import json
import glob
import cv2
import shutil
from sklearn.model_selection import train_test_split

# 1.标签路径
labelme_path = "D:\\PinInspection\\jpgretanglelabel"  # 原始labelme标注数据路径
saved_path = "D:\\PinInspection\\improveto300\\VOC2007"  # 保存路径

# 2.创建要求文件夹
dst_annotation_dir = os.path.join(saved_path, 'Annotations')
if not os.path.exists(dst_annotation_dir):
    os.makedirs(dst_annotation_dir)
dst_image_dir = os.path.join(saved_path, "JPEGImages")
if not os.path.exists(dst_image_dir):
    os.makedirs(dst_image_dir)
dst_main_dir = os.path.join(saved_path, "ImageSets", "Main")
if not os.path.exi
  • 11
    点赞
  • 44
    收藏
    觉得还不错? 一键收藏
  • 20
    评论
### 回答1: 这个错误的意思是,当n_samples=0,test_size=0.2train_size=None时,生成的训练集将为空,因此需要调整其中任意一个参数。意思是训练样本数量为0,测试集占总数据集的比例为0.2,而训练集的大小为None,导致无法生成有效的训练集。要解决这个错误,可以增加n_samples的数量,减少test_size的比例,或者为train_size指定一个值。 ### 回答2: 这个 ValueError 错误表示,在使用 sklearn 库中的 train_test_split() 函数时,我们指定的参数组合会导致训练集为空。具体来说,参数 n_samples=0 表示样本数量为0,而 test_size=0.2 表示测试集占比为20%,train_size=None 表示训练集的占比未指定。 这种情况下,函数无法从0个样本中划分出训练集,因此会引发 ValueError 错误。解决这个问题可以通过以下几个方式: 1. 增加样本数量:如果可以,我们可以增加样本数量,让样本中至少有一部分被分配到训练集中。 2. 调整测试集和训练集的占比:我们可以降低测试集的占比,或者指定训练集的占比。例如,将 test_size 设为 0.1,train_size 设为 0.9,这样训练集的占比就为90%,此时即使样本数量较少,也可以得到有效的训练集。 3. 重新选择训练集和测试集:如果实在无法调整参数,我们可以考虑重新选择训练集和测试集,从而使得训练集非空。例如,如果我们的数据集较大,可以划分出一部分样本作为训练集,剩余部分作为测试集,并将样本数量传入 train_test_split() 函数。 总之,在使用 train_test_split() 函数时,我们需要仔细考虑参数的设置,注意避免训练集为空的情况出现。如果出现了 ValueError 错误,我们可以通过调整参数、增加样本数量或重新选择训练集和测试集等方式来解决问题。 ### 回答3: 在进行机器学习模型训练时,需要将数据集划分为训练集和测试集,以便对模型进行评估和验证。在进行数据集划分时,我们通常会使用Scikit-Learn的train_test_split函数。 当使用train_test_split函数时,我们需要指定n_samples(样本数量)、test_size(测试集大小)和train_size(训练集大小)。但是,如果我们将n_samples设置为0,同时test_size设置为0.2,而train_size未设置,则会出现ValueError: with n_samples=0,test_size=0.2 and train_size=none,the resulting train set will be empty. adjust any of the aforementioned parameters.的错误提示。 这个错误提示告诉我们,由于训练集的大小未指定,而样本数量为0,因此划分后的训练集将为空。解决这个问题的方法是调整其中任何一个参数来保证划分后的训练集不为空。 例如,我们可以通过增加样本数量来解决这个问题。或者,我们可以调整测试集的大小,训练集的大小,或两者都调整,以保证划分后的训练集不为空。此外,还可以使用StratifiedShuffleSplit或其他交叉验证策略来避免这个问题。 总之,在使用train_test_split函数时,我们需要注意参数的设置,确保划分后的训练集和测试集的大小合适,并且不会出现以上的错误提示。
评论 20
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值