Update file get_train_val.py

get_train_val.py

@@ -100,7 +100,7 @@ if __name__ == '__main__':
     train, val = utils.split(df, 'train', args.ratio)
     
     saved_directory = os.path.join(args.directory, 'set')
-
+    utils.create_directory(saved_directory)
     if args.test:
         val, test = utils.split(val, 'val', 0.5)
         export_split(args, [train, val, test], args.path_to_data,