├── .gitignore ├── LICENSE ├── README.md ├── autoregressive.sh ├── baseline.sh ├── data_provider ├── data_factory.py ├── data_loader.py └── m4.py ├── exp ├── exp_basic.py └── exp_main.py ├── layers ├── AutoCorrelation.py ├── Autoformer_EncDec.py ├── Conv_Blocks.py ├── Embed.py ├── MEGA.py ├── PatchTST_backbone.py ├── PatchTST_layers.py ├── Positional_Embedding.py ├── RevIN.py ├── SelfAttention_Family.py └── Transformer_EncDec.py ├── models ├── Autoformer.py ├── Autoregressive.py ├── CATS.py ├── DLinear.py ├── FITS.py ├── FormerBaseline.py ├── Informer.py ├── Linear.py ├── NLinear.py ├── PatchTST.py ├── TiDE.py ├── TimesNet.py ├── Transformer.py └── iTransformer.py ├── requirements.txt ├── run_longExp.py └── utils ├── MoE.py ├── masking.py ├── metrics.py ├── scientific_report.py ├── timefeatures.py └── tools.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/README.md -------------------------------------------------------------------------------- /autoregressive.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/autoregressive.sh -------------------------------------------------------------------------------- /baseline.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/baseline.sh -------------------------------------------------------------------------------- /data_provider/data_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/data_provider/data_factory.py -------------------------------------------------------------------------------- /data_provider/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/data_provider/data_loader.py -------------------------------------------------------------------------------- /data_provider/m4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/data_provider/m4.py -------------------------------------------------------------------------------- /exp/exp_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/exp/exp_basic.py -------------------------------------------------------------------------------- /exp/exp_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/exp/exp_main.py -------------------------------------------------------------------------------- /layers/AutoCorrelation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/layers/AutoCorrelation.py -------------------------------------------------------------------------------- /layers/Autoformer_EncDec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/layers/Autoformer_EncDec.py -------------------------------------------------------------------------------- /layers/Conv_Blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/layers/Conv_Blocks.py -------------------------------------------------------------------------------- /layers/Embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/layers/Embed.py -------------------------------------------------------------------------------- /layers/MEGA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/layers/MEGA.py -------------------------------------------------------------------------------- /layers/PatchTST_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/layers/PatchTST_backbone.py -------------------------------------------------------------------------------- /layers/PatchTST_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/layers/PatchTST_layers.py -------------------------------------------------------------------------------- /layers/Positional_Embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/layers/Positional_Embedding.py -------------------------------------------------------------------------------- /layers/RevIN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/layers/RevIN.py -------------------------------------------------------------------------------- /layers/SelfAttention_Family.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/layers/SelfAttention_Family.py -------------------------------------------------------------------------------- /layers/Transformer_EncDec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/layers/Transformer_EncDec.py -------------------------------------------------------------------------------- /models/Autoformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/models/Autoformer.py -------------------------------------------------------------------------------- /models/Autoregressive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/models/Autoregressive.py -------------------------------------------------------------------------------- /models/CATS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/models/CATS.py -------------------------------------------------------------------------------- /models/DLinear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/models/DLinear.py -------------------------------------------------------------------------------- /models/FITS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/models/FITS.py -------------------------------------------------------------------------------- /models/FormerBaseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/models/FormerBaseline.py -------------------------------------------------------------------------------- /models/Informer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/models/Informer.py -------------------------------------------------------------------------------- /models/Linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/models/Linear.py -------------------------------------------------------------------------------- /models/NLinear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/models/NLinear.py -------------------------------------------------------------------------------- /models/PatchTST.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/models/PatchTST.py -------------------------------------------------------------------------------- /models/TiDE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/models/TiDE.py -------------------------------------------------------------------------------- /models/TimesNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/models/TimesNet.py -------------------------------------------------------------------------------- /models/Transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/models/Transformer.py -------------------------------------------------------------------------------- /models/iTransformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/models/iTransformer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_longExp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/run_longExp.py -------------------------------------------------------------------------------- /utils/MoE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/utils/MoE.py -------------------------------------------------------------------------------- /utils/masking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/utils/masking.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/scientific_report.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/utils/scientific_report.py -------------------------------------------------------------------------------- /utils/timefeatures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/utils/timefeatures.py -------------------------------------------------------------------------------- /utils/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJC-FVNR/ARMA-Attention/HEAD/utils/tools.py --------------------------------------------------------------------------------