一、前言 前几天需要划分数据集，就写了一个小demo去完成这个任务。 随机划分图片数据集 任务描述：我的所有图片保存在同一个文件夹里，需要随机将图片与标注文件划分为训练集和测试集。 处理过程：读取文件列表，将列表打乱，截取列表一部分 二、实现代码如下 三、方法补充 除了以上的方法，小编还为大家整理了其他划分数据集的方法，希望对大家有所帮助 方法一：使用random.sample将数据集随机划分为训练集与验证集并另存在文件夹中，设置随机种子 方法二：通过train test_splt函数实现随机划分数据 Python中，随机划分数据集可以通过train test_splt函数实现。该函数可以将数据集随机分成训练集和测试集，用于机器学习中的数据训练和测试。 函数的基本用法如下: 其中，X是所有的样本特征，是目标变量，test_size是测试集占总样本数的比例。random_state用于设置随机数发生器的种子，以保i每次划分结果一致。函数会返回4个数组: X_train、X_test、y_train,y_test，分别代表训练集的特征、测试集的特征、训练集的目标变量和测试集的目标变量 方法三：将一个数据集按比例随机分割成训练集、验证集、测试集 到此这篇关于Python实现随机划分图片数据集的示例代码的文章就介绍到这了,更多相关Python随机划分数据集内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！