python随机划分训练集测试集

利用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)#划分后的训练集存放位置

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