Update create_dataset.py

Author: elelo22

data/create_dataset.py

@@ -71,7 +71,7 @@ def save_to_list(self):
                                     .format(self.label_kind)), delimiter=',')
 
             # Get each original sample and create dataset samples
-            id_trials = [x.split("\\")[-1].partition("_")[0] for x in list_files(dir, sorted_dir=False)] # get beggining of files
+            id_trials = [x.split("/")[-1].partition("_")[0] for x in list_files(dir, sorted_dir=False)] # get beggining of files
             id_trials = sorted(np.unique(id_trials)[:-1], key=lambda x: int(x))  # remove duplicates, "label", and sort
             for i, id in enumerate(tqdm(id_trials, desc=f'Subject {subj}')):
                 pupil_data = np.genfromtxt(os.path.join(dir, '{}_PUPIL.csv'
@@ -258,4 +258,4 @@ def load_dataset(data, labels, scaling, noise, m, SNR):
         os.makedirs(args.save_path)
 
     torch.save(train_data, f'{args.save_path}/train_augmented_data_{args.label_kind}.pt')
-    torch.save(test_data,  f'{args.save_path}/test_data_{args.label_kind}.pt')
+    torch.save(test_data,  f'{args.save_path}/test_data_{args.label_kind}.pt')