├── .gitignore ├── LICENSE ├── README.md ├── environment.yml ├── requirements.txt └── sleepfm ├── 0_extract_pretraining_data.py ├── 1_prepare_dataset.py ├── 2_pretrain.py ├── 3_generate_embed_pretraining.py ├── 4_classification_eval_pretraining.py ├── bash_scripts ├── 0_extract_pretraining_data.sh ├── 1_prepare_dataset.sh ├── 2_pretrain.sh ├── 3_generate_embed_pretraining.sh └── 4_classification_eval_pretraining.sh ├── checkpoint └── best.pt ├── config.py ├── model ├── dataset.py └── models.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rthapa84/sleepfm-codebase/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rthapa84/sleepfm-codebase/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rthapa84/sleepfm-codebase/HEAD/README.md -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rthapa84/sleepfm-codebase/HEAD/environment.yml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rthapa84/sleepfm-codebase/HEAD/requirements.txt -------------------------------------------------------------------------------- /sleepfm/0_extract_pretraining_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rthapa84/sleepfm-codebase/HEAD/sleepfm/0_extract_pretraining_data.py -------------------------------------------------------------------------------- /sleepfm/1_prepare_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rthapa84/sleepfm-codebase/HEAD/sleepfm/1_prepare_dataset.py -------------------------------------------------------------------------------- /sleepfm/2_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rthapa84/sleepfm-codebase/HEAD/sleepfm/2_pretrain.py -------------------------------------------------------------------------------- /sleepfm/3_generate_embed_pretraining.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rthapa84/sleepfm-codebase/HEAD/sleepfm/3_generate_embed_pretraining.py -------------------------------------------------------------------------------- /sleepfm/4_classification_eval_pretraining.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rthapa84/sleepfm-codebase/HEAD/sleepfm/4_classification_eval_pretraining.py -------------------------------------------------------------------------------- /sleepfm/bash_scripts/0_extract_pretraining_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rthapa84/sleepfm-codebase/HEAD/sleepfm/bash_scripts/0_extract_pretraining_data.sh -------------------------------------------------------------------------------- /sleepfm/bash_scripts/1_prepare_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rthapa84/sleepfm-codebase/HEAD/sleepfm/bash_scripts/1_prepare_dataset.sh -------------------------------------------------------------------------------- /sleepfm/bash_scripts/2_pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rthapa84/sleepfm-codebase/HEAD/sleepfm/bash_scripts/2_pretrain.sh -------------------------------------------------------------------------------- /sleepfm/bash_scripts/3_generate_embed_pretraining.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rthapa84/sleepfm-codebase/HEAD/sleepfm/bash_scripts/3_generate_embed_pretraining.sh -------------------------------------------------------------------------------- /sleepfm/bash_scripts/4_classification_eval_pretraining.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rthapa84/sleepfm-codebase/HEAD/sleepfm/bash_scripts/4_classification_eval_pretraining.sh -------------------------------------------------------------------------------- /sleepfm/checkpoint/best.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rthapa84/sleepfm-codebase/HEAD/sleepfm/checkpoint/best.pt -------------------------------------------------------------------------------- /sleepfm/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rthapa84/sleepfm-codebase/HEAD/sleepfm/config.py -------------------------------------------------------------------------------- /sleepfm/model/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rthapa84/sleepfm-codebase/HEAD/sleepfm/model/dataset.py -------------------------------------------------------------------------------- /sleepfm/model/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rthapa84/sleepfm-codebase/HEAD/sleepfm/model/models.py -------------------------------------------------------------------------------- /sleepfm/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rthapa84/sleepfm-codebase/HEAD/sleepfm/utils.py --------------------------------------------------------------------------------