10-Python实现数据集划分（训练集/验证集/测试集）

""" Python将样本划分为训练集/验证集/测试集 """
import os, random, shutil

path = input("D:/图片原先存储路径：")
new_path = input("D:/Train_Sample存放路径：")

for root, dirs, files in os.walk(path):
    fileNumber = len(files)
    rate = 0.2
    pickNumber = int(rate * fileNumber)
    sample = random.sample(files, pickNumber)
    for name in sample:

        file_path = root + '/' + name
        new_file_path = new_path + '/' + name
        shutil.move(file_path, new_file_path)