├── .gitignore ├── Dockerfile ├── LICENSE ├── README.md ├── analysis └── plot_psd.ipynb ├── clusters ├── compute_fid_test.sh ├── compute_mssim_reconstruction.sh ├── compute_mssim_synthetic.sh ├── compute_mssim_synthetic_dm.sh ├── compute_mssim_test.sh ├── create_runai.sh ├── create_runai_preprocessing.sh ├── create_runai_sleep.sh ├── create_runai_sleep_decode.sh ├── debug_runai.sh ├── debug_runai_base_decode.sh ├── debug_runai_shhs.sh ├── delete_runai.sh ├── run_aekl.sh ├── run_aekl_jukebox.sh ├── run_aekl_jukebox_2.sh ├── run_aekl_shhs_1.sh ├── run_aekl_shhs_h.sh ├── run_base_decoder_sleep_edfx.sh ├── run_base_decoder_sleep_edfx_b.sh ├── run_base_decoder_sleep_edfx_c.sh ├── run_dm_jukebox.sh ├── run_dm_sampling_SHHS.sh ├── run_dm_sampling_edfx.sh ├── run_dm_sampling_shhs1.sh ├── run_dm_sampling_shhsh.sh ├── run_dm_shhs.sh ├── run_dm_shhs_h.sh ├── run_download.sh ├── run_infer_ldm.sh ├── run_ldm.sh ├── run_ldm_jukebox.sh ├── run_ldm_jukebox_2.sh ├── run_ldm_jukebox_2_shhs.sh ├── run_ldm_jukebox_2_shhs1.sh ├── run_ldm_jukebox_2_shhs_1.sh ├── run_ldm_jukebox_shhs_.sh ├── run_metric_shhs_1.sh ├── run_pre_processing_shhs.sh ├── run_pre_processing_sleep_edfx.sh ├── run_sampling_shhs1.sh ├── run_sampling_shhs_h.sh ├── run_sampling_sleep_edfx.sh └── suspend_runai.sh ├── config ├── config_aekl_eeg.yaml ├── config_aekl_eeg_2_2_4.yaml ├── config_aekl_eeg_2_2_4_spec.yaml ├── config_aekl_eeg_4_16_32.yaml ├── config_aekl_eeg_4_4_16.yaml ├── config_aekl_eeg_retraining.yaml ├── config_dm.yaml ├── config_ldm.yaml ├── config_sleep_stage.yaml ├── config_sleep_stage_b.yaml └── config_sleep_stage_c.yaml ├── data ├── ids │ ├── ids_sleep_edfx_cassette.csv │ ├── ids_sleep_edfx_cassette_double.csv │ ├── ids_sleep_edfx_cassette_double_test.csv │ ├── ids_sleep_edfx_cassette_double_train.csv │ ├── ids_sleep_edfx_cassette_double_valid.csv │ ├── ids_sleep_edfx_cassette_test.csv │ ├── ids_sleep_edfx_cassette_train.csv │ └── ids_sleep_edfx_cassette_valid.csv └── ids_shhs │ ├── ids_shhs.csv │ ├── ids_shhs_h.csv │ ├── ids_shhs_h_test.csv │ ├── ids_shhs_h_train.csv │ ├── ids_shhs_h_valid.csv │ ├── ids_shhs_test.csv │ ├── ids_shhs_train.csv │ ├── ids_shhs_valid.csv │ ├── ids_sleep_shhs.csv │ └── ids_sleep_shhs_h.csv ├── fig └── img.png ├── outputs ├── sleep_stage_chambon │ ├── checkpoint │ │ ├── criterion.pt │ │ ├── history.json │ │ ├── optimizer.pt │ │ └── params.pt │ ├── final_model.pth │ ├── final_model_criterion.pkl │ ├── final_model_optimizer.pkl │ ├── final_model_params.pkl │ ├── final_module_0_state.pth │ └── final_module_1_state.pth ├── sleep_stage_chambon_b │ ├── checkpoint_b │ │ ├── criterion.pt │ │ ├── history.json │ │ ├── optimizer.pt │ │ └── params.pt │ ├── confusion_matrix.png │ ├── final_model.pth │ ├── final_model_criterion.pkl │ ├── final_model_optimizer.pkl │ ├── final_model_params.pkl │ ├── final_module_0_state.pth │ └── final_module_1_state.pth ├── sleep_stage_chambon_b_old │ ├── checkpoint_b │ │ ├── criterion.pt │ │ ├── history.json │ │ ├── optimizer.pt │ │ └── params.pt │ ├── confusion_matrix.png │ ├── final_model.pth │ ├── final_model_criterion.pkl │ ├── final_model_optimizer.pkl │ ├── final_model_params.pkl │ ├── final_module_0_state.pth │ └── final_module_1_state.pth └── sleep_stage_chambon_old │ ├── checkpoint │ ├── criterion.pt │ ├── history.json │ ├── optimizer.pt │ └── params.pt │ ├── final_model.pth │ ├── final_model_criterion.pkl │ ├── final_model_history.json │ ├── final_model_optimizer.pkl │ ├── final_model_params.pkl │ ├── final_module_0_state.pth │ └── final_module_1_state.pth ├── requirements.txt └── src ├── bash └── start_training.sh ├── compute_fid.py ├── compute_fid_train_test.py ├── compute_mmds.py ├── compute_mmds_train_test.py ├── dataset └── dataset.py ├── first_version ├── train_ddpm.py └── train_encoder.py ├── models ├── ae_kl.py ├── discriminator.py ├── ldm.py ├── unet.py └── usleep.py ├── preprocessing ├── convert_edfx.py ├── convert_shhs.py ├── read_monai.py ├── split_train_valid_test_sleep_edfx.py └── split_train_valid_test_sleep_shhs.py ├── sample_trials.py ├── sample_trials_ddpm.py ├── testing ├── FID_test.py ├── FID_train_test.py ├── MSSIM_dm_sample.py ├── MSSIM_reconstruction.py ├── MSSIM_sample.py ├── MSSIM_test.py ├── MSSIM_test_shhs.py ├── compute_fid.py ├── compute_fid_train_test.py ├── compute_mmds.py ├── compute_mmds_train_test.py ├── computer_fid_based_feature.py ├── computer_fid_based_feature_train.py ├── computer_spectrum.py ├── computer_spectrum_aekl.py ├── convert_mlflow_to_pytorch.py ├── dataset.py ├── metric_scores.py ├── run_sleep_decode.py ├── run_sleep_decode_b.py ├── run_sleep_decode_c.py ├── sample_trials.py ├── sample_trials_autoencoder.py ├── sample_trials_ddpm.py ├── sample_trials_test.py ├── tailored_mssim.py ├── tailored_ssim.py ├── torch_ssim.py └── usleep.py ├── train_autoencoderkl.py ├── train_ldm.py ├── train_pure_ldm.py ├── training ├── __init__.py ├── training.py ├── training_diffusion.py └── util.py └── util.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/.gitignore -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/README.md -------------------------------------------------------------------------------- /analysis/plot_psd.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/analysis/plot_psd.ipynb -------------------------------------------------------------------------------- /clusters/compute_fid_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/compute_fid_test.sh -------------------------------------------------------------------------------- /clusters/compute_mssim_reconstruction.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/compute_mssim_reconstruction.sh -------------------------------------------------------------------------------- /clusters/compute_mssim_synthetic.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/compute_mssim_synthetic.sh -------------------------------------------------------------------------------- /clusters/compute_mssim_synthetic_dm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/compute_mssim_synthetic_dm.sh -------------------------------------------------------------------------------- /clusters/compute_mssim_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/compute_mssim_test.sh -------------------------------------------------------------------------------- /clusters/create_runai.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/create_runai.sh -------------------------------------------------------------------------------- /clusters/create_runai_preprocessing.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/create_runai_preprocessing.sh -------------------------------------------------------------------------------- /clusters/create_runai_sleep.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/create_runai_sleep.sh -------------------------------------------------------------------------------- /clusters/create_runai_sleep_decode.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/create_runai_sleep_decode.sh -------------------------------------------------------------------------------- /clusters/debug_runai.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/debug_runai.sh -------------------------------------------------------------------------------- /clusters/debug_runai_base_decode.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/debug_runai_base_decode.sh -------------------------------------------------------------------------------- /clusters/debug_runai_shhs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/debug_runai_shhs.sh -------------------------------------------------------------------------------- /clusters/delete_runai.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/delete_runai.sh -------------------------------------------------------------------------------- /clusters/run_aekl.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_aekl.sh -------------------------------------------------------------------------------- /clusters/run_aekl_jukebox.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_aekl_jukebox.sh -------------------------------------------------------------------------------- /clusters/run_aekl_jukebox_2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_aekl_jukebox_2.sh -------------------------------------------------------------------------------- /clusters/run_aekl_shhs_1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_aekl_shhs_1.sh -------------------------------------------------------------------------------- /clusters/run_aekl_shhs_h.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_aekl_shhs_h.sh -------------------------------------------------------------------------------- /clusters/run_base_decoder_sleep_edfx.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_base_decoder_sleep_edfx.sh -------------------------------------------------------------------------------- /clusters/run_base_decoder_sleep_edfx_b.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_base_decoder_sleep_edfx_b.sh -------------------------------------------------------------------------------- /clusters/run_base_decoder_sleep_edfx_c.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_base_decoder_sleep_edfx_c.sh -------------------------------------------------------------------------------- /clusters/run_dm_jukebox.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_dm_jukebox.sh -------------------------------------------------------------------------------- /clusters/run_dm_sampling_SHHS.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_dm_sampling_SHHS.sh -------------------------------------------------------------------------------- /clusters/run_dm_sampling_edfx.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_dm_sampling_edfx.sh -------------------------------------------------------------------------------- /clusters/run_dm_sampling_shhs1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_dm_sampling_shhs1.sh -------------------------------------------------------------------------------- /clusters/run_dm_sampling_shhsh.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_dm_sampling_shhsh.sh -------------------------------------------------------------------------------- /clusters/run_dm_shhs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_dm_shhs.sh -------------------------------------------------------------------------------- /clusters/run_dm_shhs_h.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_dm_shhs_h.sh -------------------------------------------------------------------------------- /clusters/run_download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_download.sh -------------------------------------------------------------------------------- /clusters/run_infer_ldm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_infer_ldm.sh -------------------------------------------------------------------------------- /clusters/run_ldm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_ldm.sh -------------------------------------------------------------------------------- /clusters/run_ldm_jukebox.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_ldm_jukebox.sh -------------------------------------------------------------------------------- /clusters/run_ldm_jukebox_2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_ldm_jukebox_2.sh -------------------------------------------------------------------------------- /clusters/run_ldm_jukebox_2_shhs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_ldm_jukebox_2_shhs.sh -------------------------------------------------------------------------------- /clusters/run_ldm_jukebox_2_shhs1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_ldm_jukebox_2_shhs1.sh -------------------------------------------------------------------------------- /clusters/run_ldm_jukebox_2_shhs_1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_ldm_jukebox_2_shhs_1.sh -------------------------------------------------------------------------------- /clusters/run_ldm_jukebox_shhs_.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_ldm_jukebox_shhs_.sh -------------------------------------------------------------------------------- /clusters/run_metric_shhs_1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_metric_shhs_1.sh -------------------------------------------------------------------------------- /clusters/run_pre_processing_shhs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_pre_processing_shhs.sh -------------------------------------------------------------------------------- /clusters/run_pre_processing_sleep_edfx.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_pre_processing_sleep_edfx.sh -------------------------------------------------------------------------------- /clusters/run_sampling_shhs1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_sampling_shhs1.sh -------------------------------------------------------------------------------- /clusters/run_sampling_shhs_h.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_sampling_shhs_h.sh -------------------------------------------------------------------------------- /clusters/run_sampling_sleep_edfx.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/run_sampling_sleep_edfx.sh -------------------------------------------------------------------------------- /clusters/suspend_runai.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/clusters/suspend_runai.sh -------------------------------------------------------------------------------- /config/config_aekl_eeg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/config/config_aekl_eeg.yaml -------------------------------------------------------------------------------- /config/config_aekl_eeg_2_2_4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/config/config_aekl_eeg_2_2_4.yaml -------------------------------------------------------------------------------- /config/config_aekl_eeg_2_2_4_spec.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/config/config_aekl_eeg_2_2_4_spec.yaml -------------------------------------------------------------------------------- /config/config_aekl_eeg_4_16_32.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/config/config_aekl_eeg_4_16_32.yaml -------------------------------------------------------------------------------- /config/config_aekl_eeg_4_4_16.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/config/config_aekl_eeg_4_4_16.yaml -------------------------------------------------------------------------------- /config/config_aekl_eeg_retraining.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/config/config_aekl_eeg_retraining.yaml -------------------------------------------------------------------------------- /config/config_dm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/config/config_dm.yaml -------------------------------------------------------------------------------- /config/config_ldm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/config/config_ldm.yaml -------------------------------------------------------------------------------- /config/config_sleep_stage.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/config/config_sleep_stage.yaml -------------------------------------------------------------------------------- /config/config_sleep_stage_b.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/config/config_sleep_stage_b.yaml -------------------------------------------------------------------------------- /config/config_sleep_stage_c.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/config/config_sleep_stage_c.yaml -------------------------------------------------------------------------------- /data/ids/ids_sleep_edfx_cassette.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/data/ids/ids_sleep_edfx_cassette.csv -------------------------------------------------------------------------------- /data/ids/ids_sleep_edfx_cassette_double.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/data/ids/ids_sleep_edfx_cassette_double.csv -------------------------------------------------------------------------------- /data/ids/ids_sleep_edfx_cassette_double_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/data/ids/ids_sleep_edfx_cassette_double_test.csv -------------------------------------------------------------------------------- /data/ids/ids_sleep_edfx_cassette_double_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/data/ids/ids_sleep_edfx_cassette_double_train.csv -------------------------------------------------------------------------------- /data/ids/ids_sleep_edfx_cassette_double_valid.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/data/ids/ids_sleep_edfx_cassette_double_valid.csv -------------------------------------------------------------------------------- /data/ids/ids_sleep_edfx_cassette_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/data/ids/ids_sleep_edfx_cassette_test.csv -------------------------------------------------------------------------------- /data/ids/ids_sleep_edfx_cassette_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/data/ids/ids_sleep_edfx_cassette_train.csv -------------------------------------------------------------------------------- /data/ids/ids_sleep_edfx_cassette_valid.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/data/ids/ids_sleep_edfx_cassette_valid.csv -------------------------------------------------------------------------------- /data/ids_shhs/ids_shhs.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/data/ids_shhs/ids_shhs.csv -------------------------------------------------------------------------------- /data/ids_shhs/ids_shhs_h.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/data/ids_shhs/ids_shhs_h.csv -------------------------------------------------------------------------------- /data/ids_shhs/ids_shhs_h_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/data/ids_shhs/ids_shhs_h_test.csv -------------------------------------------------------------------------------- /data/ids_shhs/ids_shhs_h_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/data/ids_shhs/ids_shhs_h_train.csv -------------------------------------------------------------------------------- /data/ids_shhs/ids_shhs_h_valid.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/data/ids_shhs/ids_shhs_h_valid.csv -------------------------------------------------------------------------------- /data/ids_shhs/ids_shhs_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/data/ids_shhs/ids_shhs_test.csv -------------------------------------------------------------------------------- /data/ids_shhs/ids_shhs_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/data/ids_shhs/ids_shhs_train.csv -------------------------------------------------------------------------------- /data/ids_shhs/ids_shhs_valid.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/data/ids_shhs/ids_shhs_valid.csv -------------------------------------------------------------------------------- /data/ids_shhs/ids_sleep_shhs.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/data/ids_shhs/ids_sleep_shhs.csv -------------------------------------------------------------------------------- /data/ids_shhs/ids_sleep_shhs_h.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/data/ids_shhs/ids_sleep_shhs_h.csv -------------------------------------------------------------------------------- /fig/img.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/fig/img.png -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon/checkpoint/criterion.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon/checkpoint/criterion.pt -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon/checkpoint/history.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon/checkpoint/history.json -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon/checkpoint/optimizer.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon/checkpoint/optimizer.pt -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon/checkpoint/params.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon/checkpoint/params.pt -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon/final_model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon/final_model.pth -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon/final_model_criterion.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon/final_model_criterion.pkl -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon/final_model_optimizer.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon/final_model_optimizer.pkl -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon/final_model_params.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon/final_model_params.pkl -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon/final_module_0_state.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon/final_module_0_state.pth -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon/final_module_1_state.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon/final_module_1_state.pth -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_b/checkpoint_b/criterion.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_b/checkpoint_b/criterion.pt -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_b/checkpoint_b/history.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_b/checkpoint_b/history.json -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_b/checkpoint_b/optimizer.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_b/checkpoint_b/optimizer.pt -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_b/checkpoint_b/params.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_b/checkpoint_b/params.pt -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_b/confusion_matrix.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_b/confusion_matrix.png -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_b/final_model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_b/final_model.pth -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_b/final_model_criterion.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_b/final_model_criterion.pkl -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_b/final_model_optimizer.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_b/final_model_optimizer.pkl -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_b/final_model_params.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_b/final_model_params.pkl -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_b/final_module_0_state.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_b/final_module_0_state.pth -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_b/final_module_1_state.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_b/final_module_1_state.pth -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_b_old/checkpoint_b/criterion.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_b_old/checkpoint_b/criterion.pt -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_b_old/checkpoint_b/history.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_b_old/checkpoint_b/history.json -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_b_old/checkpoint_b/optimizer.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_b_old/checkpoint_b/optimizer.pt -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_b_old/checkpoint_b/params.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_b_old/checkpoint_b/params.pt -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_b_old/confusion_matrix.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_b_old/confusion_matrix.png -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_b_old/final_model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_b_old/final_model.pth -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_b_old/final_model_criterion.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_b_old/final_model_criterion.pkl -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_b_old/final_model_optimizer.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_b_old/final_model_optimizer.pkl -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_b_old/final_model_params.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_b_old/final_model_params.pkl -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_b_old/final_module_0_state.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_b_old/final_module_0_state.pth -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_b_old/final_module_1_state.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_b_old/final_module_1_state.pth -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_old/checkpoint/criterion.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_old/checkpoint/criterion.pt -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_old/checkpoint/history.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_old/checkpoint/history.json -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_old/checkpoint/optimizer.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_old/checkpoint/optimizer.pt -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_old/checkpoint/params.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_old/checkpoint/params.pt -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_old/final_model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_old/final_model.pth -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_old/final_model_criterion.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_old/final_model_criterion.pkl -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_old/final_model_history.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_old/final_model_history.json -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_old/final_model_optimizer.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_old/final_model_optimizer.pkl -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_old/final_model_params.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_old/final_model_params.pkl -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_old/final_module_0_state.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_old/final_module_0_state.pth -------------------------------------------------------------------------------- /outputs/sleep_stage_chambon_old/final_module_1_state.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/outputs/sleep_stage_chambon_old/final_module_1_state.pth -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/bash/start_training.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/bash/start_training.sh -------------------------------------------------------------------------------- /src/compute_fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/compute_fid.py -------------------------------------------------------------------------------- /src/compute_fid_train_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/compute_fid_train_test.py -------------------------------------------------------------------------------- /src/compute_mmds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/compute_mmds.py -------------------------------------------------------------------------------- /src/compute_mmds_train_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/compute_mmds_train_test.py -------------------------------------------------------------------------------- /src/dataset/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/dataset/dataset.py -------------------------------------------------------------------------------- /src/first_version/train_ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/first_version/train_ddpm.py -------------------------------------------------------------------------------- /src/first_version/train_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/first_version/train_encoder.py -------------------------------------------------------------------------------- /src/models/ae_kl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/models/ae_kl.py -------------------------------------------------------------------------------- /src/models/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/models/discriminator.py -------------------------------------------------------------------------------- /src/models/ldm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/models/ldm.py -------------------------------------------------------------------------------- /src/models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/models/unet.py -------------------------------------------------------------------------------- /src/models/usleep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/models/usleep.py -------------------------------------------------------------------------------- /src/preprocessing/convert_edfx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/preprocessing/convert_edfx.py -------------------------------------------------------------------------------- /src/preprocessing/convert_shhs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/preprocessing/convert_shhs.py -------------------------------------------------------------------------------- /src/preprocessing/read_monai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/preprocessing/read_monai.py -------------------------------------------------------------------------------- /src/preprocessing/split_train_valid_test_sleep_edfx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/preprocessing/split_train_valid_test_sleep_edfx.py -------------------------------------------------------------------------------- /src/preprocessing/split_train_valid_test_sleep_shhs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/preprocessing/split_train_valid_test_sleep_shhs.py -------------------------------------------------------------------------------- /src/sample_trials.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/sample_trials.py -------------------------------------------------------------------------------- /src/sample_trials_ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/sample_trials_ddpm.py -------------------------------------------------------------------------------- /src/testing/FID_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/FID_test.py -------------------------------------------------------------------------------- /src/testing/FID_train_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/FID_train_test.py -------------------------------------------------------------------------------- /src/testing/MSSIM_dm_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/MSSIM_dm_sample.py -------------------------------------------------------------------------------- /src/testing/MSSIM_reconstruction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/MSSIM_reconstruction.py -------------------------------------------------------------------------------- /src/testing/MSSIM_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/MSSIM_sample.py -------------------------------------------------------------------------------- /src/testing/MSSIM_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/MSSIM_test.py -------------------------------------------------------------------------------- /src/testing/MSSIM_test_shhs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/MSSIM_test_shhs.py -------------------------------------------------------------------------------- /src/testing/compute_fid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/compute_fid.py -------------------------------------------------------------------------------- /src/testing/compute_fid_train_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/compute_fid_train_test.py -------------------------------------------------------------------------------- /src/testing/compute_mmds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/compute_mmds.py -------------------------------------------------------------------------------- /src/testing/compute_mmds_train_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/compute_mmds_train_test.py -------------------------------------------------------------------------------- /src/testing/computer_fid_based_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/computer_fid_based_feature.py -------------------------------------------------------------------------------- /src/testing/computer_fid_based_feature_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/computer_fid_based_feature_train.py -------------------------------------------------------------------------------- /src/testing/computer_spectrum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/computer_spectrum.py -------------------------------------------------------------------------------- /src/testing/computer_spectrum_aekl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/computer_spectrum_aekl.py -------------------------------------------------------------------------------- /src/testing/convert_mlflow_to_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/convert_mlflow_to_pytorch.py -------------------------------------------------------------------------------- /src/testing/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/dataset.py -------------------------------------------------------------------------------- /src/testing/metric_scores.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/metric_scores.py -------------------------------------------------------------------------------- /src/testing/run_sleep_decode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/run_sleep_decode.py -------------------------------------------------------------------------------- /src/testing/run_sleep_decode_b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/run_sleep_decode_b.py -------------------------------------------------------------------------------- /src/testing/run_sleep_decode_c.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/run_sleep_decode_c.py -------------------------------------------------------------------------------- /src/testing/sample_trials.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/sample_trials.py -------------------------------------------------------------------------------- /src/testing/sample_trials_autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/sample_trials_autoencoder.py -------------------------------------------------------------------------------- /src/testing/sample_trials_ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/sample_trials_ddpm.py -------------------------------------------------------------------------------- /src/testing/sample_trials_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/sample_trials_test.py -------------------------------------------------------------------------------- /src/testing/tailored_mssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/tailored_mssim.py -------------------------------------------------------------------------------- /src/testing/tailored_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/tailored_ssim.py -------------------------------------------------------------------------------- /src/testing/torch_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/torch_ssim.py -------------------------------------------------------------------------------- /src/testing/usleep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/testing/usleep.py -------------------------------------------------------------------------------- /src/train_autoencoderkl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/train_autoencoderkl.py -------------------------------------------------------------------------------- /src/train_ldm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/train_ldm.py -------------------------------------------------------------------------------- /src/train_pure_ldm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/train_pure_ldm.py -------------------------------------------------------------------------------- /src/training/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/training/__init__.py -------------------------------------------------------------------------------- /src/training/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/training/training.py -------------------------------------------------------------------------------- /src/training/training_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/training/training_diffusion.py -------------------------------------------------------------------------------- /src/training/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/training/util.py -------------------------------------------------------------------------------- /src/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bruAristimunha/Synthetic-Sleep-EEG-Signal-Generation-using-Latent-Diffusion-Models/HEAD/src/util.py --------------------------------------------------------------------------------