python随机划分训练集测试集

tech_98

已于 2023-03-19 21:55:41 修改

阅读量912

点赞数 2

文章标签： python 机器学习深度学习人工智能 Powered by 金山文档

于 2023-03-06 21:54:00 首次发布

本文链接：https://blog.csdn.net/bianbian111111/article/details/129371725

版权

利用python划分训练集和测试集方法如下

from sklearn.model_selection import train_test_split
from PIL import Image
import os


path="./path/your_datasets" #你要划分的数据集

dat= os.listdir(path)

train, test = train_test_split(dat, test_size = 0.2)#0.2表示需要划分的测试集比例

for file in train:

    im2 = Image.open("./path/your_datasets"+file)
    im2.save("./path/train/"+file)#划分后的训练集存放位置

for file in test:

    im2 = Image.open("./path/your_datasets"+file)
    im2.save("./path/test/"+file)#划分后的训练集存放位置

根据注释改成自己的路径，运行结束就可以得到自己随机划分的训练集和测试集了。