File size: 978 Bytes
8f8716a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 | from config import *
# Uni-encoder models (simclr, dino, mae, vqvae, ar, etc.)
TRAIN_EDF_COLS_UNI_ENC = [ECG, EMG_Chin, EMG_LLeg, EMG_RLeg,
ABD, THX, NP, SN,
EOG_E1_A2, EOG_E2_A1,EEG_C3_A2, EEG_C4_A1,
]
TRAIN_EDF_COLS_MULTI_ENC = [ECG,
ABD, THX, NP, SN,
EMG_Chin, EMG_LLeg, EMG_RLeg,
EOG_E1_A2, EOG_E2_A1,EEG_C3_A2, EEG_C4_A1,
]
TRAIN_EDF_COLS_TYPE3 = [ECG, ABD, THX, NP, SN]
TRAIN_EDF_COLS_TYPE4 = [ECG, ABD, THX]
MONITOR_TYPE_MAP = {
"main": TRAIN_EDF_COLS_UNI_ENC,
"type3": TRAIN_EDF_COLS_TYPE3,
"type4": TRAIN_EDF_COLS_TYPE4,
}
STAGE2_LABEL_PATH_WITH_PATHHEAD = "/path/to/your/label/splits"
CKPT_PATH = "/path/to/your/checkpoints"
MODEL_LIST = ["dino_ours"]
AUGMENTATION_MAP = {
"dino_ours": "chan_then_pcspan",
}
SPLIT_DATA_FOLDER = "/path/to/your/postprocessed/data"
PRETRAIN_VAL_DATASET_LIST = ['shhs']
NEED_NORM_COL = [HR, SPO2, OX] |