Train_test split now moves both train and test files

tools/split_data_in_train_test.py

@@ -42,12 +42,15 @@ def main():
     df = pd.read_csv(args.ground_truth)
     df.dropna(inplace=True)
     # Set aside 10% for hyperparameter optimisation:
-    _, set_aside_for_hyperparam = train_test_split(df, test_size=args.test_size,
+    train_data, test_data = train_test_split(df, test_size=args.test_size,
                                              stratify=df['species'])
-    set_aside_files = list(set_aside_for_hyperparam['file name'])
+    train_files = list(train_data['file name'])
+    test_files = list(test_data['file name'])
 
-    move_files_from_list(set_aside_files, in_directory=args.in_dir,
-                         out_directory=args.out_dir)
+    move_files_from_list(train_files, in_directory=args.in_dir,
+                         out_directory=os.path.join(args.out_dir, 'train'))
+    move_files_from_list(test_files, in_directory=args.in_dir,
+                         out_directory=os.path.join(args.out_dir, 'test'))
 
 
 if __name__ == '__main__':