├── .gitignore ├── LICENSE ├── README.md ├── checkpoints └── download_checkpoints.py ├── configs ├── SleePyCo-Transformer_SL-01_numScales-1_MASS_pretrain.json ├── SleePyCo-Transformer_SL-01_numScales-1_Physio2018_pretrain.json ├── SleePyCo-Transformer_SL-01_numScales-1_SHHS_pretrain.json ├── SleePyCo-Transformer_SL-01_numScales-1_Sleep-EDF-2013_pretrain.json ├── SleePyCo-Transformer_SL-01_numScales-1_Sleep-EDF-2018_pretrain.json ├── SleePyCo-Transformer_SL-10_numScales-3_MASS_freezefinetune.json ├── SleePyCo-Transformer_SL-10_numScales-3_Physio2018_freezefinetune.json ├── SleePyCo-Transformer_SL-10_numScales-3_SHHS_freezefinetune.json ├── SleePyCo-Transformer_SL-10_numScales-3_Sleep-EDF-2013_freezefinetune.json ├── SleePyCo-Transformer_SL-10_numScales-3_Sleep-EDF-2018_freezefinetune.json ├── ablations_backbone │ ├── DeepSleepNet-Transformer_SL-01_numScales-1_MASS_pretrain.json │ ├── DeepSleepNet-Transformer_SL-01_numScales-1_Physio2018_pretrain.json │ ├── DeepSleepNet-Transformer_SL-01_numScales-1_SHHS_pretrain.json │ ├── DeepSleepNet-Transformer_SL-01_numScales-1_Sleep-EDF-2018_pretrain.json │ ├── DeepSleepNet-Transformer_SL-10_numScales-1_MASS_freezefinetune.json │ ├── DeepSleepNet-Transformer_SL-10_numScales-1_Physio2018_freezefinetune.json │ ├── DeepSleepNet-Transformer_SL-10_numScales-1_SHHS_freezefinetune.json │ ├── DeepSleepNet-Transformer_SL-10_numScales-1_Sleep-EDF-2018_freezefinetune.json │ ├── IITNet-Transformer_SL-01_numScales-1_MASS_pretrain.json │ ├── IITNet-Transformer_SL-01_numScales-1_Physio2018_pretrain.json │ ├── IITNet-Transformer_SL-01_numScales-1_SHHS_pretrain.json │ ├── IITNet-Transformer_SL-01_numScales-1_Sleep-EDF-2018_pretrain.json │ ├── IITNet-Transformer_SL-10_numScales-1_MASS_freezefinetune.json │ ├── IITNet-Transformer_SL-10_numScales-1_Physio2018_freezefinetune.json │ ├── IITNet-Transformer_SL-10_numScales-1_SHHS_freezefinetune.json │ ├── IITNet-Transformer_SL-10_numScales-1_Sleep-EDF-2018_freezefinetune.json │ ├── TinySleepNet-Transformer_SL-01_numScales-1_MASS_pretrain.json │ ├── TinySleepNet-Transformer_SL-01_numScales-1_Physio2018_pretrain.json │ ├── TinySleepNet-Transformer_SL-01_numScales-1_SHHS_pretrain.json │ ├── TinySleepNet-Transformer_SL-01_numScales-1_Sleep-EDF-2018_pretrain.json │ ├── TinySleepNet-Transformer_SL-10_numScales-1_MASS_freezefinetune.json │ ├── TinySleepNet-Transformer_SL-10_numScales-1_Physio2018_freezefinetune.json │ ├── TinySleepNet-Transformer_SL-10_numScales-1_SHHS_freezefinetune.json │ ├── TinySleepNet-Transformer_SL-10_numScales-1_Sleep-EDF-2018_freezefinetune.json │ ├── UTime-Transformer_SL-01_numScales-1_MASS_pretrain.json │ ├── UTime-Transformer_SL-01_numScales-1_Physio2018_pretrain.json │ ├── UTime-Transformer_SL-01_numScales-1_SHHS_pretrain.json │ ├── UTime-Transformer_SL-01_numScales-1_Sleep-EDF-2018_pretrain.json │ ├── UTime-Transformer_SL-10_numScales-1_MASS_freezefinetune.json │ ├── UTime-Transformer_SL-10_numScales-1_Physio2018_freezefinetune.json │ ├── UTime-Transformer_SL-10_numScales-1_SHHS_freezefinetune.json │ ├── UTime-Transformer_SL-10_numScales-1_Sleep-EDF-2018_freezefinetune.json │ ├── UTime-Transformer_SL-10_numScales-3_MASS_freezefinetune.json │ ├── UTime-Transformer_SL-10_numScales-3_Physio2018_freezefinetune.json │ ├── UTime-Transformer_SL-10_numScales-3_SHHS_freezefinetune.json │ ├── UTime-Transformer_SL-10_numScales-3_Sleep-EDF-2018_freezefinetune.json │ ├── XSleepNet-Transformer_SL-01_numScales-1_MASS_pretrain.json │ ├── XSleepNet-Transformer_SL-01_numScales-1_Physio2018_pretrain.json │ ├── XSleepNet-Transformer_SL-01_numScales-1_SHHS_pretrain.json │ ├── XSleepNet-Transformer_SL-01_numScales-1_Sleep-EDF-2018_pretrain.json │ ├── XSleepNet-Transformer_SL-10_numScales-1_MASS_freezefinetune.json │ ├── XSleepNet-Transformer_SL-10_numScales-1_Physio2018_freezefinetune.json │ ├── XSleepNet-Transformer_SL-10_numScales-1_SHHS_freezefinetune.json │ ├── XSleepNet-Transformer_SL-10_numScales-1_Sleep-EDF-2018_freezefinetune.json │ ├── XSleepNet-Transformer_SL-10_numScales-3_MASS_freezefinetune.json │ ├── XSleepNet-Transformer_SL-10_numScales-3_Physio2018_freezefinetune.json │ ├── XSleepNet-Transformer_SL-10_numScales-3_SHHS_freezefinetune.json │ └── XSleepNet-Transformer_SL-10_numScales-3_Sleep-EDF-2018_freezefinetune.json └── ablations_component │ ├── SleePyCo-Transformer_SL-01_numScales-1_Sleep-EDF-2018_pretrain.json │ ├── SleePyCo-Transformer_SL-10_numScales-1_Sleep-EDF-2018_freezefinetune.json │ ├── SleePyCo-Transformer_SL-10_numScales-1_Sleep-EDF-2018_scratch.json │ ├── SleePyCo-Transformer_SL-10_numScales-3_Sleep-EDF-2018_freezefinetune.json │ └── SleePyCo-Transformer_SL-10_numScales-3_Sleep-EDF-2018_scratch.json ├── dset ├── Sleep-EDF-2013 │ ├── download_sleep-edf-2013.py │ └── prepare_sleep-edf-2013.py └── Sleep-EDF-2018 │ ├── download_sleep-edf-2018.py │ └── prepare_sleep-edf-2018.py ├── figures ├── framework.png └── model.png ├── loader.py ├── loss.py ├── models ├── classifiers.py ├── deepsleepnet.py ├── iitnet.py ├── main_model.py ├── sleepyco.py ├── tinysleepnet.py ├── utils.py ├── utime.py └── xsleepnet.py ├── preprocess_utils ├── README.md ├── prepare_mass.py ├── prepare_physio2018.py ├── prepare_shhs.py ├── prepare_sleepedf.py └── sleepstage.py ├── requirements.txt ├── split_idx ├── idx_MASS.npy ├── idx_Physio2018.npy ├── idx_SHHS.npy ├── idx_Sleep-EDF-2013.npy └── idx_Sleep-EDF-2018.npy ├── test.py ├── test_custom.py ├── tools ├── README.md ├── draw_hypnogram_2x1.py ├── draw_hypnogram_2x2.py └── source │ ├── 02-6_pred.npy │ ├── 02-6_true.npy │ ├── 07-5_pred.npy │ ├── 07-5_true.npy │ ├── hypnogram_2x1.png │ └── hypnogram_2x2.png ├── train_crl.py ├── train_mtcl.py ├── transform.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/README.md -------------------------------------------------------------------------------- /checkpoints/download_checkpoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/checkpoints/download_checkpoints.py -------------------------------------------------------------------------------- /configs/SleePyCo-Transformer_SL-01_numScales-1_MASS_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/SleePyCo-Transformer_SL-01_numScales-1_MASS_pretrain.json -------------------------------------------------------------------------------- /configs/SleePyCo-Transformer_SL-01_numScales-1_Physio2018_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/SleePyCo-Transformer_SL-01_numScales-1_Physio2018_pretrain.json -------------------------------------------------------------------------------- /configs/SleePyCo-Transformer_SL-01_numScales-1_SHHS_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/SleePyCo-Transformer_SL-01_numScales-1_SHHS_pretrain.json -------------------------------------------------------------------------------- /configs/SleePyCo-Transformer_SL-01_numScales-1_Sleep-EDF-2013_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/SleePyCo-Transformer_SL-01_numScales-1_Sleep-EDF-2013_pretrain.json -------------------------------------------------------------------------------- /configs/SleePyCo-Transformer_SL-01_numScales-1_Sleep-EDF-2018_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/SleePyCo-Transformer_SL-01_numScales-1_Sleep-EDF-2018_pretrain.json -------------------------------------------------------------------------------- /configs/SleePyCo-Transformer_SL-10_numScales-3_MASS_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/SleePyCo-Transformer_SL-10_numScales-3_MASS_freezefinetune.json -------------------------------------------------------------------------------- /configs/SleePyCo-Transformer_SL-10_numScales-3_Physio2018_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/SleePyCo-Transformer_SL-10_numScales-3_Physio2018_freezefinetune.json -------------------------------------------------------------------------------- /configs/SleePyCo-Transformer_SL-10_numScales-3_SHHS_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/SleePyCo-Transformer_SL-10_numScales-3_SHHS_freezefinetune.json -------------------------------------------------------------------------------- /configs/SleePyCo-Transformer_SL-10_numScales-3_Sleep-EDF-2013_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/SleePyCo-Transformer_SL-10_numScales-3_Sleep-EDF-2013_freezefinetune.json -------------------------------------------------------------------------------- /configs/SleePyCo-Transformer_SL-10_numScales-3_Sleep-EDF-2018_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/SleePyCo-Transformer_SL-10_numScales-3_Sleep-EDF-2018_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/DeepSleepNet-Transformer_SL-01_numScales-1_MASS_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/DeepSleepNet-Transformer_SL-01_numScales-1_MASS_pretrain.json -------------------------------------------------------------------------------- /configs/ablations_backbone/DeepSleepNet-Transformer_SL-01_numScales-1_Physio2018_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/DeepSleepNet-Transformer_SL-01_numScales-1_Physio2018_pretrain.json -------------------------------------------------------------------------------- /configs/ablations_backbone/DeepSleepNet-Transformer_SL-01_numScales-1_SHHS_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/DeepSleepNet-Transformer_SL-01_numScales-1_SHHS_pretrain.json -------------------------------------------------------------------------------- /configs/ablations_backbone/DeepSleepNet-Transformer_SL-01_numScales-1_Sleep-EDF-2018_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/DeepSleepNet-Transformer_SL-01_numScales-1_Sleep-EDF-2018_pretrain.json -------------------------------------------------------------------------------- /configs/ablations_backbone/DeepSleepNet-Transformer_SL-10_numScales-1_MASS_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/DeepSleepNet-Transformer_SL-10_numScales-1_MASS_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/DeepSleepNet-Transformer_SL-10_numScales-1_Physio2018_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/DeepSleepNet-Transformer_SL-10_numScales-1_Physio2018_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/DeepSleepNet-Transformer_SL-10_numScales-1_SHHS_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/DeepSleepNet-Transformer_SL-10_numScales-1_SHHS_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/DeepSleepNet-Transformer_SL-10_numScales-1_Sleep-EDF-2018_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/DeepSleepNet-Transformer_SL-10_numScales-1_Sleep-EDF-2018_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/IITNet-Transformer_SL-01_numScales-1_MASS_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/IITNet-Transformer_SL-01_numScales-1_MASS_pretrain.json -------------------------------------------------------------------------------- /configs/ablations_backbone/IITNet-Transformer_SL-01_numScales-1_Physio2018_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/IITNet-Transformer_SL-01_numScales-1_Physio2018_pretrain.json -------------------------------------------------------------------------------- /configs/ablations_backbone/IITNet-Transformer_SL-01_numScales-1_SHHS_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/IITNet-Transformer_SL-01_numScales-1_SHHS_pretrain.json -------------------------------------------------------------------------------- /configs/ablations_backbone/IITNet-Transformer_SL-01_numScales-1_Sleep-EDF-2018_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/IITNet-Transformer_SL-01_numScales-1_Sleep-EDF-2018_pretrain.json -------------------------------------------------------------------------------- /configs/ablations_backbone/IITNet-Transformer_SL-10_numScales-1_MASS_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/IITNet-Transformer_SL-10_numScales-1_MASS_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/IITNet-Transformer_SL-10_numScales-1_Physio2018_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/IITNet-Transformer_SL-10_numScales-1_Physio2018_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/IITNet-Transformer_SL-10_numScales-1_SHHS_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/IITNet-Transformer_SL-10_numScales-1_SHHS_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/IITNet-Transformer_SL-10_numScales-1_Sleep-EDF-2018_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/IITNet-Transformer_SL-10_numScales-1_Sleep-EDF-2018_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/TinySleepNet-Transformer_SL-01_numScales-1_MASS_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/TinySleepNet-Transformer_SL-01_numScales-1_MASS_pretrain.json -------------------------------------------------------------------------------- /configs/ablations_backbone/TinySleepNet-Transformer_SL-01_numScales-1_Physio2018_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/TinySleepNet-Transformer_SL-01_numScales-1_Physio2018_pretrain.json -------------------------------------------------------------------------------- /configs/ablations_backbone/TinySleepNet-Transformer_SL-01_numScales-1_SHHS_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/TinySleepNet-Transformer_SL-01_numScales-1_SHHS_pretrain.json -------------------------------------------------------------------------------- /configs/ablations_backbone/TinySleepNet-Transformer_SL-01_numScales-1_Sleep-EDF-2018_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/TinySleepNet-Transformer_SL-01_numScales-1_Sleep-EDF-2018_pretrain.json -------------------------------------------------------------------------------- /configs/ablations_backbone/TinySleepNet-Transformer_SL-10_numScales-1_MASS_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/TinySleepNet-Transformer_SL-10_numScales-1_MASS_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/TinySleepNet-Transformer_SL-10_numScales-1_Physio2018_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/TinySleepNet-Transformer_SL-10_numScales-1_Physio2018_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/TinySleepNet-Transformer_SL-10_numScales-1_SHHS_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/TinySleepNet-Transformer_SL-10_numScales-1_SHHS_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/TinySleepNet-Transformer_SL-10_numScales-1_Sleep-EDF-2018_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/TinySleepNet-Transformer_SL-10_numScales-1_Sleep-EDF-2018_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/UTime-Transformer_SL-01_numScales-1_MASS_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/UTime-Transformer_SL-01_numScales-1_MASS_pretrain.json -------------------------------------------------------------------------------- /configs/ablations_backbone/UTime-Transformer_SL-01_numScales-1_Physio2018_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/UTime-Transformer_SL-01_numScales-1_Physio2018_pretrain.json -------------------------------------------------------------------------------- /configs/ablations_backbone/UTime-Transformer_SL-01_numScales-1_SHHS_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/UTime-Transformer_SL-01_numScales-1_SHHS_pretrain.json -------------------------------------------------------------------------------- /configs/ablations_backbone/UTime-Transformer_SL-01_numScales-1_Sleep-EDF-2018_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/UTime-Transformer_SL-01_numScales-1_Sleep-EDF-2018_pretrain.json -------------------------------------------------------------------------------- /configs/ablations_backbone/UTime-Transformer_SL-10_numScales-1_MASS_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/UTime-Transformer_SL-10_numScales-1_MASS_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/UTime-Transformer_SL-10_numScales-1_Physio2018_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/UTime-Transformer_SL-10_numScales-1_Physio2018_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/UTime-Transformer_SL-10_numScales-1_SHHS_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/UTime-Transformer_SL-10_numScales-1_SHHS_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/UTime-Transformer_SL-10_numScales-1_Sleep-EDF-2018_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/UTime-Transformer_SL-10_numScales-1_Sleep-EDF-2018_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/UTime-Transformer_SL-10_numScales-3_MASS_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/UTime-Transformer_SL-10_numScales-3_MASS_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/UTime-Transformer_SL-10_numScales-3_Physio2018_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/UTime-Transformer_SL-10_numScales-3_Physio2018_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/UTime-Transformer_SL-10_numScales-3_SHHS_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/UTime-Transformer_SL-10_numScales-3_SHHS_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/UTime-Transformer_SL-10_numScales-3_Sleep-EDF-2018_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/UTime-Transformer_SL-10_numScales-3_Sleep-EDF-2018_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/XSleepNet-Transformer_SL-01_numScales-1_MASS_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/XSleepNet-Transformer_SL-01_numScales-1_MASS_pretrain.json -------------------------------------------------------------------------------- /configs/ablations_backbone/XSleepNet-Transformer_SL-01_numScales-1_Physio2018_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/XSleepNet-Transformer_SL-01_numScales-1_Physio2018_pretrain.json -------------------------------------------------------------------------------- /configs/ablations_backbone/XSleepNet-Transformer_SL-01_numScales-1_SHHS_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/XSleepNet-Transformer_SL-01_numScales-1_SHHS_pretrain.json -------------------------------------------------------------------------------- /configs/ablations_backbone/XSleepNet-Transformer_SL-01_numScales-1_Sleep-EDF-2018_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/XSleepNet-Transformer_SL-01_numScales-1_Sleep-EDF-2018_pretrain.json -------------------------------------------------------------------------------- /configs/ablations_backbone/XSleepNet-Transformer_SL-10_numScales-1_MASS_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/XSleepNet-Transformer_SL-10_numScales-1_MASS_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/XSleepNet-Transformer_SL-10_numScales-1_Physio2018_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/XSleepNet-Transformer_SL-10_numScales-1_Physio2018_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/XSleepNet-Transformer_SL-10_numScales-1_SHHS_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/XSleepNet-Transformer_SL-10_numScales-1_SHHS_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/XSleepNet-Transformer_SL-10_numScales-1_Sleep-EDF-2018_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/XSleepNet-Transformer_SL-10_numScales-1_Sleep-EDF-2018_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/XSleepNet-Transformer_SL-10_numScales-3_MASS_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/XSleepNet-Transformer_SL-10_numScales-3_MASS_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/XSleepNet-Transformer_SL-10_numScales-3_Physio2018_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/XSleepNet-Transformer_SL-10_numScales-3_Physio2018_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/XSleepNet-Transformer_SL-10_numScales-3_SHHS_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/XSleepNet-Transformer_SL-10_numScales-3_SHHS_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_backbone/XSleepNet-Transformer_SL-10_numScales-3_Sleep-EDF-2018_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_backbone/XSleepNet-Transformer_SL-10_numScales-3_Sleep-EDF-2018_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_component/SleePyCo-Transformer_SL-01_numScales-1_Sleep-EDF-2018_pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_component/SleePyCo-Transformer_SL-01_numScales-1_Sleep-EDF-2018_pretrain.json -------------------------------------------------------------------------------- /configs/ablations_component/SleePyCo-Transformer_SL-10_numScales-1_Sleep-EDF-2018_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_component/SleePyCo-Transformer_SL-10_numScales-1_Sleep-EDF-2018_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_component/SleePyCo-Transformer_SL-10_numScales-1_Sleep-EDF-2018_scratch.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_component/SleePyCo-Transformer_SL-10_numScales-1_Sleep-EDF-2018_scratch.json -------------------------------------------------------------------------------- /configs/ablations_component/SleePyCo-Transformer_SL-10_numScales-3_Sleep-EDF-2018_freezefinetune.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_component/SleePyCo-Transformer_SL-10_numScales-3_Sleep-EDF-2018_freezefinetune.json -------------------------------------------------------------------------------- /configs/ablations_component/SleePyCo-Transformer_SL-10_numScales-3_Sleep-EDF-2018_scratch.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/configs/ablations_component/SleePyCo-Transformer_SL-10_numScales-3_Sleep-EDF-2018_scratch.json -------------------------------------------------------------------------------- /dset/Sleep-EDF-2013/download_sleep-edf-2013.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/dset/Sleep-EDF-2013/download_sleep-edf-2013.py -------------------------------------------------------------------------------- /dset/Sleep-EDF-2013/prepare_sleep-edf-2013.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/dset/Sleep-EDF-2013/prepare_sleep-edf-2013.py -------------------------------------------------------------------------------- /dset/Sleep-EDF-2018/download_sleep-edf-2018.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/dset/Sleep-EDF-2018/download_sleep-edf-2018.py -------------------------------------------------------------------------------- /dset/Sleep-EDF-2018/prepare_sleep-edf-2018.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/dset/Sleep-EDF-2018/prepare_sleep-edf-2018.py -------------------------------------------------------------------------------- /figures/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/figures/framework.png -------------------------------------------------------------------------------- /figures/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/figures/model.png -------------------------------------------------------------------------------- /loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/loader.py -------------------------------------------------------------------------------- /loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/loss.py -------------------------------------------------------------------------------- /models/classifiers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/models/classifiers.py -------------------------------------------------------------------------------- /models/deepsleepnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/models/deepsleepnet.py -------------------------------------------------------------------------------- /models/iitnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/models/iitnet.py -------------------------------------------------------------------------------- /models/main_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/models/main_model.py -------------------------------------------------------------------------------- /models/sleepyco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/models/sleepyco.py -------------------------------------------------------------------------------- /models/tinysleepnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/models/tinysleepnet.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/models/utils.py -------------------------------------------------------------------------------- /models/utime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/models/utime.py -------------------------------------------------------------------------------- /models/xsleepnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/models/xsleepnet.py -------------------------------------------------------------------------------- /preprocess_utils/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/preprocess_utils/README.md -------------------------------------------------------------------------------- /preprocess_utils/prepare_mass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/preprocess_utils/prepare_mass.py -------------------------------------------------------------------------------- /preprocess_utils/prepare_physio2018.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/preprocess_utils/prepare_physio2018.py -------------------------------------------------------------------------------- /preprocess_utils/prepare_shhs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/preprocess_utils/prepare_shhs.py -------------------------------------------------------------------------------- /preprocess_utils/prepare_sleepedf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/preprocess_utils/prepare_sleepedf.py -------------------------------------------------------------------------------- /preprocess_utils/sleepstage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/preprocess_utils/sleepstage.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/requirements.txt -------------------------------------------------------------------------------- /split_idx/idx_MASS.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/split_idx/idx_MASS.npy -------------------------------------------------------------------------------- /split_idx/idx_Physio2018.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/split_idx/idx_Physio2018.npy -------------------------------------------------------------------------------- /split_idx/idx_SHHS.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/split_idx/idx_SHHS.npy -------------------------------------------------------------------------------- /split_idx/idx_Sleep-EDF-2013.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/split_idx/idx_Sleep-EDF-2013.npy -------------------------------------------------------------------------------- /split_idx/idx_Sleep-EDF-2018.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/split_idx/idx_Sleep-EDF-2018.npy -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/test.py -------------------------------------------------------------------------------- /test_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/test_custom.py -------------------------------------------------------------------------------- /tools/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/tools/README.md -------------------------------------------------------------------------------- /tools/draw_hypnogram_2x1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/tools/draw_hypnogram_2x1.py -------------------------------------------------------------------------------- /tools/draw_hypnogram_2x2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/tools/draw_hypnogram_2x2.py -------------------------------------------------------------------------------- /tools/source/02-6_pred.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/tools/source/02-6_pred.npy -------------------------------------------------------------------------------- /tools/source/02-6_true.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/tools/source/02-6_true.npy -------------------------------------------------------------------------------- /tools/source/07-5_pred.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/tools/source/07-5_pred.npy -------------------------------------------------------------------------------- /tools/source/07-5_true.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/tools/source/07-5_true.npy -------------------------------------------------------------------------------- /tools/source/hypnogram_2x1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/tools/source/hypnogram_2x1.png -------------------------------------------------------------------------------- /tools/source/hypnogram_2x2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/tools/source/hypnogram_2x2.png -------------------------------------------------------------------------------- /train_crl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/train_crl.py -------------------------------------------------------------------------------- /train_mtcl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/train_mtcl.py -------------------------------------------------------------------------------- /transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/transform.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gist-ailab/SleePyCo/HEAD/utils.py --------------------------------------------------------------------------------