automate_training: split_dataset

Author: charleygros

ivadomed/scripts/automate_training.py

@@ -180,18 +180,18 @@ def automate_training(config, param, fixed_split, all_combin, n_iterations=1, ru
 
     # Split dataset if not already done
     if fixed_split and (initial_config.get("split_path") is None):
-        train_lst, valid_lst, test_lst = imed_loader_utils.split_dataset(path_folder=initial_config["bids_path"],
-                                                                         center_test_lst=initial_config["center_test"],
-                                                                         split_method=initial_config["split_method"],
-                                                                         random_seed=initial_config["random_seed"],
-                                                                         train_frac=initial_config["train_fraction"],
-                                                                         test_frac=initial_config["test_fraction"])
+        train_lst, valid_lst, test_lst = imed_loader_utils.split_dataset(path_folder=initial_config["loader_parameters"]["bids_path"],
+                                                                         center_test_lst=initial_config["split_dataset"]["center_test"],
+                                                                         split_method=initial_config["split_dataset"]["split_method"],
+                                                                         random_seed=initial_config["split_dataset"]["random_seed"],
+                                                                         train_frac=initial_config["split_dataset"]["train_fraction"],
+                                                                         test_frac=initial_config["split_dataset"]["test_fraction"])
 
         # save the subject distribution
         split_dct = {'train': train_lst, 'valid': valid_lst, 'test': test_lst}
         split_path = "./" + "common_split_datasets.joblib"
         joblib.dump(split_dct, split_path)
-        initial_config["split_path"] = split_path
+        initial_config["split_dataset"]["fname_split"] = split_path
 
     config_list = []
     # Test all combinations (change multiple parameters for each test)