Generate a Tree for Train/Dev/Test with folders for each class

gen_datatree.py

#!/usr/bin/env python

import os
import argparse

FOLDERS = ['train', 'dev', 'test']

parser = argparse.ArgumentParser(
    description='Folder strcture generator for datasets')
parser.add_argument('classes', type=str,
                    help='List of classes separated by columns')

args = parser.parse_args()
classes = args.classes
each_class = classes.split(',')

paths = []

for folder in FOLDERS:
    for cl in each_class:
        paths.append(os.path.join(folder, cl))
try:
    for path in paths:
        os.makedirs(path)
except OSError as e:
    print('Could not create all folders:', e, sep='\n')