├── .gitignore ├── LICENSE ├── README.md ├── data_provider ├── __init__.py ├── data_factory.py └── data_loader.py ├── exp ├── __init__.py ├── exp_basic.py └── exp_main.py ├── layers ├── Causal_Conv.py ├── Corrformer_EncDec.py ├── Embed.py ├── Multi_Correlation.py └── __init__.py ├── models ├── Corrformer.py └── __init__.py ├── requirements.txt ├── run.py ├── scripts ├── Demo │ ├── Global_Temp_demo.sh │ └── Global_Wind_demo.sh ├── Global_Temp │ └── Corrformer.sh └── Global_Wind │ └── Corrformer.sh └── utils ├── __init__.py ├── masking.py ├── metrics.py ├── timefeatures.py └── tools.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Corrformer/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Corrformer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Corrformer/HEAD/README.md -------------------------------------------------------------------------------- /data_provider/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /data_provider/data_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Corrformer/HEAD/data_provider/data_factory.py -------------------------------------------------------------------------------- /data_provider/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Corrformer/HEAD/data_provider/data_loader.py -------------------------------------------------------------------------------- /exp/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /exp/exp_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Corrformer/HEAD/exp/exp_basic.py -------------------------------------------------------------------------------- /exp/exp_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Corrformer/HEAD/exp/exp_main.py -------------------------------------------------------------------------------- /layers/Causal_Conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Corrformer/HEAD/layers/Causal_Conv.py -------------------------------------------------------------------------------- /layers/Corrformer_EncDec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Corrformer/HEAD/layers/Corrformer_EncDec.py -------------------------------------------------------------------------------- /layers/Embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Corrformer/HEAD/layers/Embed.py -------------------------------------------------------------------------------- /layers/Multi_Correlation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Corrformer/HEAD/layers/Multi_Correlation.py -------------------------------------------------------------------------------- /layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/Corrformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Corrformer/HEAD/models/Corrformer.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Corrformer/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Corrformer/HEAD/run.py -------------------------------------------------------------------------------- /scripts/Demo/Global_Temp_demo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Corrformer/HEAD/scripts/Demo/Global_Temp_demo.sh -------------------------------------------------------------------------------- /scripts/Demo/Global_Wind_demo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Corrformer/HEAD/scripts/Demo/Global_Wind_demo.sh -------------------------------------------------------------------------------- /scripts/Global_Temp/Corrformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Corrformer/HEAD/scripts/Global_Temp/Corrformer.sh -------------------------------------------------------------------------------- /scripts/Global_Wind/Corrformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Corrformer/HEAD/scripts/Global_Wind/Corrformer.sh -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/masking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Corrformer/HEAD/utils/masking.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Corrformer/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/timefeatures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Corrformer/HEAD/utils/timefeatures.py -------------------------------------------------------------------------------- /utils/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/Corrformer/HEAD/utils/tools.py --------------------------------------------------------------------------------