training Organizing files in train/test directories For a prototype model I'm writing that involves a multi-class