如何在python中将数据集的4个文件夹拆分到不同的文件夹？

Question

我正在尝试使用python进行形状图像分类的CNN教程之一。我有4个不同形状的文件夹。我尝试使用以下代码拆分数据集。

os.mkdir(os.path.join(path_target, 'train'))
os.mkdir(os.path.join(path_target, 'valid'))
for t in ['train', 'valid']:
    for folder in ['circle/', 'square/', 'star/', 'triangle/']:
        os.mkdir(os.path.join(path_target, t, folder))

def preprocessData(dirName, ext):
    allFiles = list()
    for root, dirs, files in os.walk(dirName):
        for file in files:
            if file.endswith(ext):
                allFiles.append(os.path.join(root, file))
    Shuffle(allFiles)Split = 0.7
    split_index = floor(len(allFiles) * split)
    training = allFiles[:split_index]
    testing = allFiles[split_index:]
    return training, testing

但是我得到了以下错误

  File "<ipython-input-29-6be535540069>", line 8
    Shuffle(allFiles)Split =0.7
                         ^
SyntaxError: invalid syntax

在python中有没有办法拆分数据集文件夹？

Answer 1

import random    

def preprocessData(dirName, ext):
    allFiles = list()
    for root, dirs, files in os.walk(dirName):
        for file in files:
            if file.endswith(ext):
                allFiles.append(os.path.join(root, file))
    random.shuffle(allFiles)
    split = 0.7
    split_index = floor(len(allFiles) * split)
    training = allFiles[:split_index]
    testing = allFiles[split_index:]
    return training, testing