diff --git a/SHHS_process.py b/dataset_builder/HYS_dataset.py similarity index 100% rename from SHHS_process.py rename to dataset_builder/HYS_dataset.py diff --git a/dataset_builder/__init__.py b/dataset_builder/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/HYS_process.py b/event_mask_process/HYS_process.py similarity index 98% rename from HYS_process.py rename to event_mask_process/HYS_process.py index 06156eb..1054dba 100644 --- a/HYS_process.py +++ b/event_mask_process/HYS_process.py @@ -160,7 +160,7 @@ def process_one_signal(samp_id, show=False): movement_list=resp_movement_position_list, sampling_rate=resp_fs, **resp_amp_change_conf, - verbose=True) + verbose=False) print( f"amp_change_mask_shape: {resp_amp_change_mask.shape}, num_amp_change: {np.sum(resp_amp_change_mask == 1)}, count_amp_change_positions: {len(resp_amp_change_list)}") else: @@ -259,8 +259,8 @@ def process_one_signal(samp_id, show=False): if __name__ == '__main__': - yaml_path = Path("./dataset_config/HYS_config.yaml") - disable_df_path = Path("./排除区间.xlsx") + yaml_path = Path("../dataset_config/HYS_config.yaml") + disable_df_path = Path("../排除区间.xlsx") conf = utils.load_dataset_conf(yaml_path) select_ids = conf["select_ids"] @@ -276,8 +276,8 @@ if __name__ == '__main__': all_samp_disable_df = utils.read_disable_excel(disable_df_path) - process_one_signal(select_ids[9], show=True) - + process_one_signal(select_ids[6], show=True) + # # for samp_id in select_ids: # print(f"Processing sample ID: {samp_id}") # process_one_signal(samp_id, show=False) diff --git a/event_mask_process/SHHS_process.py b/event_mask_process/SHHS_process.py new file mode 100644 index 0000000..e69de29 diff --git a/ZD5Y_process.py b/event_mask_process/ZD5Y_process.py similarity index 99% rename from ZD5Y_process.py rename to event_mask_process/ZD5Y_process.py index 23dc4c8..8701d65 100644 --- a/ZD5Y_process.py +++ b/event_mask_process/ZD5Y_process.py @@ -252,8 +252,8 @@ def process_one_signal(samp_id, show=False): if __name__ == '__main__': - yaml_path = Path("./dataset_config/ZD5Y_config.yaml") - disable_df_path = Path("./排除区间.xlsx") + yaml_path = Path("../dataset_config/ZD5Y_config.yaml") + disable_df_path = Path("../排除区间.xlsx") conf = utils.load_dataset_conf(yaml_path) select_ids = conf["select_ids"] diff --git a/event_mask_process/__init__.py b/event_mask_process/__init__.py new file mode 100644 index 0000000..e69de29