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)