├── .github ├── ISSUE_TEMPLATE │ ├── bug-report.yml │ ├── config.yml │ └── other-issues.yml ├── PULL_REQUEST_TEMPLATE.md ├── dependabot.yml └── workflows │ ├── greetings.yml │ └── stale.yml ├── CITATION.cff ├── Global_Config.py ├── LICENSE ├── NNI_tuning ├── BRITS │ ├── BRITS_basic_config.ini │ ├── BRITS_searching_config.yml │ └── BRITS_searching_space.json ├── MRNN │ ├── MRNN_basic_config.ini │ ├── MRNN_searching_config.yml │ └── MRNN_searching_space.json ├── SAITS │ ├── SAITS_basic_config.ini │ ├── SAITS_searching_config.yml │ └── SAITS_searching_space.json └── Transformer │ ├── Transformer_basic_config.ini │ ├── Transformer_searching_config.yml │ └── Transformer_searching_space.json ├── Paper_SAITS.pdf ├── README.md ├── Simple_RNN_on_imputed_data.py ├── conda_env_dependencies.yml ├── configs ├── AirQuality_BRITS_best.ini ├── AirQuality_MRNN_best.ini ├── AirQuality_SAITS_base.ini ├── AirQuality_SAITS_best.ini ├── AirQuality_Transformer_best.ini ├── Electricity_BRITS_best.ini ├── Electricity_MRNN_best.ini ├── Electricity_SAITS_base.ini ├── Electricity_SAITS_best.ini ├── Electricity_Transformer_best.ini ├── NRTSI_AirQuality_SAITS_best.ini ├── NRTSI_Gas_SAITS_best.ini ├── PhysioNet2012_BRITS_best.ini ├── PhysioNet2012_MRNN_best.ini ├── PhysioNet2012_SAITS_base.ini ├── PhysioNet2012_SAITS_best.ini └── PhysioNet2012_Transformer_best.ini ├── dataset_generating_scripts ├── README.md ├── data_downloading.sh ├── data_processing_utils.py ├── dataset_generating.sh ├── gene_ETTm1_dataset.py ├── gene_NRTSI_dataset.py ├── gene_PhysioNet2012_dataset.py ├── gene_UCI_BeijingAirQuality_dataset.py └── gene_UCI_electricity_dataset.py ├── figs ├── SAITS arch.svg └── Training approach.svg ├── modeling ├── brits.py ├── layers.py ├── mrnn.py ├── saits.py ├── transformer.py ├── unified_dataloader.py └── utils.py └── run_models.py /.github/ISSUE_TEMPLATE/bug-report.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/.github/ISSUE_TEMPLATE/bug-report.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/.github/ISSUE_TEMPLATE/config.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/other-issues.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/.github/ISSUE_TEMPLATE/other-issues.yml -------------------------------------------------------------------------------- /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/.github/PULL_REQUEST_TEMPLATE.md -------------------------------------------------------------------------------- /.github/dependabot.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/.github/dependabot.yml -------------------------------------------------------------------------------- /.github/workflows/greetings.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/.github/workflows/greetings.yml -------------------------------------------------------------------------------- /.github/workflows/stale.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/.github/workflows/stale.yml -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/CITATION.cff -------------------------------------------------------------------------------- /Global_Config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/Global_Config.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/LICENSE -------------------------------------------------------------------------------- /NNI_tuning/BRITS/BRITS_basic_config.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/NNI_tuning/BRITS/BRITS_basic_config.ini -------------------------------------------------------------------------------- /NNI_tuning/BRITS/BRITS_searching_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/NNI_tuning/BRITS/BRITS_searching_config.yml -------------------------------------------------------------------------------- /NNI_tuning/BRITS/BRITS_searching_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/NNI_tuning/BRITS/BRITS_searching_space.json -------------------------------------------------------------------------------- /NNI_tuning/MRNN/MRNN_basic_config.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/NNI_tuning/MRNN/MRNN_basic_config.ini -------------------------------------------------------------------------------- /NNI_tuning/MRNN/MRNN_searching_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/NNI_tuning/MRNN/MRNN_searching_config.yml -------------------------------------------------------------------------------- /NNI_tuning/MRNN/MRNN_searching_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/NNI_tuning/MRNN/MRNN_searching_space.json -------------------------------------------------------------------------------- /NNI_tuning/SAITS/SAITS_basic_config.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/NNI_tuning/SAITS/SAITS_basic_config.ini -------------------------------------------------------------------------------- /NNI_tuning/SAITS/SAITS_searching_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/NNI_tuning/SAITS/SAITS_searching_config.yml -------------------------------------------------------------------------------- /NNI_tuning/SAITS/SAITS_searching_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/NNI_tuning/SAITS/SAITS_searching_space.json -------------------------------------------------------------------------------- /NNI_tuning/Transformer/Transformer_basic_config.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/NNI_tuning/Transformer/Transformer_basic_config.ini -------------------------------------------------------------------------------- /NNI_tuning/Transformer/Transformer_searching_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/NNI_tuning/Transformer/Transformer_searching_config.yml -------------------------------------------------------------------------------- /NNI_tuning/Transformer/Transformer_searching_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/NNI_tuning/Transformer/Transformer_searching_space.json -------------------------------------------------------------------------------- /Paper_SAITS.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/Paper_SAITS.pdf -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/README.md -------------------------------------------------------------------------------- /Simple_RNN_on_imputed_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/Simple_RNN_on_imputed_data.py -------------------------------------------------------------------------------- /conda_env_dependencies.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/conda_env_dependencies.yml -------------------------------------------------------------------------------- /configs/AirQuality_BRITS_best.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/configs/AirQuality_BRITS_best.ini -------------------------------------------------------------------------------- /configs/AirQuality_MRNN_best.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/configs/AirQuality_MRNN_best.ini -------------------------------------------------------------------------------- /configs/AirQuality_SAITS_base.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/configs/AirQuality_SAITS_base.ini -------------------------------------------------------------------------------- /configs/AirQuality_SAITS_best.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/configs/AirQuality_SAITS_best.ini -------------------------------------------------------------------------------- /configs/AirQuality_Transformer_best.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/configs/AirQuality_Transformer_best.ini -------------------------------------------------------------------------------- /configs/Electricity_BRITS_best.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/configs/Electricity_BRITS_best.ini -------------------------------------------------------------------------------- /configs/Electricity_MRNN_best.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/configs/Electricity_MRNN_best.ini -------------------------------------------------------------------------------- /configs/Electricity_SAITS_base.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/configs/Electricity_SAITS_base.ini -------------------------------------------------------------------------------- /configs/Electricity_SAITS_best.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/configs/Electricity_SAITS_best.ini -------------------------------------------------------------------------------- /configs/Electricity_Transformer_best.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/configs/Electricity_Transformer_best.ini -------------------------------------------------------------------------------- /configs/NRTSI_AirQuality_SAITS_best.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/configs/NRTSI_AirQuality_SAITS_best.ini -------------------------------------------------------------------------------- /configs/NRTSI_Gas_SAITS_best.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/configs/NRTSI_Gas_SAITS_best.ini -------------------------------------------------------------------------------- /configs/PhysioNet2012_BRITS_best.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/configs/PhysioNet2012_BRITS_best.ini -------------------------------------------------------------------------------- /configs/PhysioNet2012_MRNN_best.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/configs/PhysioNet2012_MRNN_best.ini -------------------------------------------------------------------------------- /configs/PhysioNet2012_SAITS_base.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/configs/PhysioNet2012_SAITS_base.ini -------------------------------------------------------------------------------- /configs/PhysioNet2012_SAITS_best.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/configs/PhysioNet2012_SAITS_best.ini -------------------------------------------------------------------------------- /configs/PhysioNet2012_Transformer_best.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/configs/PhysioNet2012_Transformer_best.ini -------------------------------------------------------------------------------- /dataset_generating_scripts/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/dataset_generating_scripts/README.md -------------------------------------------------------------------------------- /dataset_generating_scripts/data_downloading.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/dataset_generating_scripts/data_downloading.sh -------------------------------------------------------------------------------- /dataset_generating_scripts/data_processing_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/dataset_generating_scripts/data_processing_utils.py -------------------------------------------------------------------------------- /dataset_generating_scripts/dataset_generating.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/dataset_generating_scripts/dataset_generating.sh -------------------------------------------------------------------------------- /dataset_generating_scripts/gene_ETTm1_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/dataset_generating_scripts/gene_ETTm1_dataset.py -------------------------------------------------------------------------------- /dataset_generating_scripts/gene_NRTSI_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/dataset_generating_scripts/gene_NRTSI_dataset.py -------------------------------------------------------------------------------- /dataset_generating_scripts/gene_PhysioNet2012_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/dataset_generating_scripts/gene_PhysioNet2012_dataset.py -------------------------------------------------------------------------------- /dataset_generating_scripts/gene_UCI_BeijingAirQuality_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/dataset_generating_scripts/gene_UCI_BeijingAirQuality_dataset.py -------------------------------------------------------------------------------- /dataset_generating_scripts/gene_UCI_electricity_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/dataset_generating_scripts/gene_UCI_electricity_dataset.py -------------------------------------------------------------------------------- /figs/SAITS arch.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/figs/SAITS arch.svg -------------------------------------------------------------------------------- /figs/Training approach.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/figs/Training approach.svg -------------------------------------------------------------------------------- /modeling/brits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/modeling/brits.py -------------------------------------------------------------------------------- /modeling/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/modeling/layers.py -------------------------------------------------------------------------------- /modeling/mrnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/modeling/mrnn.py -------------------------------------------------------------------------------- /modeling/saits.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/modeling/saits.py -------------------------------------------------------------------------------- /modeling/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/modeling/transformer.py -------------------------------------------------------------------------------- /modeling/unified_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/modeling/unified_dataloader.py -------------------------------------------------------------------------------- /modeling/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/modeling/utils.py -------------------------------------------------------------------------------- /run_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/WenjieDu/SAITS/HEAD/run_models.py --------------------------------------------------------------------------------