├── .gitignore ├── LICENSE ├── README.md ├── dataset ├── __init__.py ├── data_info.py ├── data_parser.py └── utils.py ├── downstream ├── __init__.py ├── fine_tuning │ ├── __init__.py │ ├── data_loader.py │ ├── model.py │ └── train.py ├── linear_probing │ ├── __init__.py │ ├── data_loader.py │ ├── model.py │ └── train.py └── utils.py ├── figures └── architecture.jpg ├── models ├── __init__.py ├── loss.py ├── neuronet │ ├── __init__.py │ ├── encoder.py │ └── model.py ├── unified │ ├── __init__.py │ └── model.py └── utils.py ├── pretrained ├── __init__.py ├── multimodal │ ├── __init__.py │ ├── data_loader.py │ └── train.py └── unimodal │ ├── __init__.py │ ├── ecg │ ├── __init__.py │ ├── data_loader.py │ └── train.py │ ├── eeg │ ├── __init__.py │ ├── data_loader.py │ └── train.py │ ├── emg │ ├── __init__.py │ ├── data_loader.py │ └── train.py │ └── eog │ ├── __init__.py │ ├── data_loader.py │ └── train.py └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/README.md -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/data_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/dataset/data_info.py -------------------------------------------------------------------------------- /dataset/data_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/dataset/data_parser.py -------------------------------------------------------------------------------- /dataset/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/dataset/utils.py -------------------------------------------------------------------------------- /downstream/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /downstream/fine_tuning/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /downstream/fine_tuning/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/downstream/fine_tuning/data_loader.py -------------------------------------------------------------------------------- /downstream/fine_tuning/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/downstream/fine_tuning/model.py -------------------------------------------------------------------------------- /downstream/fine_tuning/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/downstream/fine_tuning/train.py -------------------------------------------------------------------------------- /downstream/linear_probing/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /downstream/linear_probing/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/downstream/linear_probing/data_loader.py -------------------------------------------------------------------------------- /downstream/linear_probing/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/downstream/linear_probing/model.py -------------------------------------------------------------------------------- /downstream/linear_probing/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/downstream/linear_probing/train.py -------------------------------------------------------------------------------- /downstream/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/downstream/utils.py -------------------------------------------------------------------------------- /figures/architecture.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/figures/architecture.jpg -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/models/loss.py -------------------------------------------------------------------------------- /models/neuronet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/neuronet/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/models/neuronet/encoder.py -------------------------------------------------------------------------------- /models/neuronet/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/models/neuronet/model.py -------------------------------------------------------------------------------- /models/unified/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/unified/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/models/unified/model.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/models/utils.py -------------------------------------------------------------------------------- /pretrained/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretrained/multimodal/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretrained/multimodal/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/pretrained/multimodal/data_loader.py -------------------------------------------------------------------------------- /pretrained/multimodal/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/pretrained/multimodal/train.py -------------------------------------------------------------------------------- /pretrained/unimodal/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretrained/unimodal/ecg/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretrained/unimodal/ecg/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/pretrained/unimodal/ecg/data_loader.py -------------------------------------------------------------------------------- /pretrained/unimodal/ecg/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/pretrained/unimodal/ecg/train.py -------------------------------------------------------------------------------- /pretrained/unimodal/eeg/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretrained/unimodal/eeg/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/pretrained/unimodal/eeg/data_loader.py -------------------------------------------------------------------------------- /pretrained/unimodal/eeg/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/pretrained/unimodal/eeg/train.py -------------------------------------------------------------------------------- /pretrained/unimodal/emg/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretrained/unimodal/emg/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/pretrained/unimodal/emg/data_loader.py -------------------------------------------------------------------------------- /pretrained/unimodal/emg/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/pretrained/unimodal/emg/train.py -------------------------------------------------------------------------------- /pretrained/unimodal/eog/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretrained/unimodal/eog/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/pretrained/unimodal/eog/data_loader.py -------------------------------------------------------------------------------- /pretrained/unimodal/eog/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/pretrained/unimodal/eog/train.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dlcjfgmlnasa/SynthSleepNet/HEAD/requirements.txt --------------------------------------------------------------------------------