├── .gitattributes ├── .gitignore ├── ACF_ETTh1.ipynb ├── Figures ├── Figure5.png ├── Figure6.png ├── Model.png ├── Sparse.png ├── Table2.png ├── Table4.png └── Table7.png ├── LICENSE ├── README.md ├── data_provider ├── data_factory.py └── data_loader.py ├── exp ├── exp_basic.py └── exp_main.py ├── layers ├── AutoCorrelation.py ├── Autoformer_EncDec.py ├── Embed.py ├── PatchTST_backbone.py ├── PatchTST_layers.py ├── RevIN.py ├── SelfAttention_Family.py ├── Transformer_EncDec.py └── utils.py ├── models ├── Autoformer.py ├── DLinear.py ├── FEDformer.py ├── Film.py ├── Informer.py ├── Linear.py ├── PatchTST.py ├── SparseTSF.py ├── Stat_models.py └── Transformer.py ├── requirements.txt ├── run_all.sh ├── run_longExp.py ├── scripts └── SparseTSF │ ├── linear-S │ ├── electricity.sh │ ├── etth1.sh │ ├── etth2.sh │ ├── ettm1.sh │ ├── ettm2.sh │ ├── solar.sh │ ├── traffic.sh │ └── weather.sh │ ├── linear │ ├── electricity.sh │ ├── etth1.sh │ ├── etth2.sh │ ├── ettm1.sh │ ├── ettm2.sh │ ├── solar.sh │ ├── traffic.sh │ └── weather.sh │ ├── mlp-S │ ├── electricity.sh │ ├── etth1.sh │ ├── etth2.sh │ ├── ettm1.sh │ ├── ettm2.sh │ ├── solar.sh │ ├── traffic.sh │ └── weather.sh │ └── mlp │ ├── electricity.sh │ ├── etth1.sh │ ├── etth2.sh │ ├── ettm1.sh │ ├── ettm2.sh │ ├── solar.sh │ ├── traffic.sh │ └── weather.sh └── utils ├── augmentations.py ├── masking.py ├── metrics.py ├── timefeatures.py └── tools.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/.gitignore -------------------------------------------------------------------------------- /ACF_ETTh1.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/ACF_ETTh1.ipynb -------------------------------------------------------------------------------- /Figures/Figure5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/Figures/Figure5.png -------------------------------------------------------------------------------- /Figures/Figure6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/Figures/Figure6.png -------------------------------------------------------------------------------- /Figures/Model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/Figures/Model.png -------------------------------------------------------------------------------- /Figures/Sparse.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/Figures/Sparse.png -------------------------------------------------------------------------------- /Figures/Table2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/Figures/Table2.png -------------------------------------------------------------------------------- /Figures/Table4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/Figures/Table4.png -------------------------------------------------------------------------------- /Figures/Table7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/Figures/Table7.png -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/README.md -------------------------------------------------------------------------------- /data_provider/data_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/data_provider/data_factory.py -------------------------------------------------------------------------------- /data_provider/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/data_provider/data_loader.py -------------------------------------------------------------------------------- /exp/exp_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/exp/exp_basic.py -------------------------------------------------------------------------------- /exp/exp_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/exp/exp_main.py -------------------------------------------------------------------------------- /layers/AutoCorrelation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/layers/AutoCorrelation.py -------------------------------------------------------------------------------- /layers/Autoformer_EncDec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/layers/Autoformer_EncDec.py -------------------------------------------------------------------------------- /layers/Embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/layers/Embed.py -------------------------------------------------------------------------------- /layers/PatchTST_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/layers/PatchTST_backbone.py -------------------------------------------------------------------------------- /layers/PatchTST_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/layers/PatchTST_layers.py -------------------------------------------------------------------------------- /layers/RevIN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/layers/RevIN.py -------------------------------------------------------------------------------- /layers/SelfAttention_Family.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/layers/SelfAttention_Family.py -------------------------------------------------------------------------------- /layers/Transformer_EncDec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/layers/Transformer_EncDec.py -------------------------------------------------------------------------------- /layers/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/layers/utils.py -------------------------------------------------------------------------------- /models/Autoformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/models/Autoformer.py -------------------------------------------------------------------------------- /models/DLinear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/models/DLinear.py -------------------------------------------------------------------------------- /models/FEDformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/models/FEDformer.py -------------------------------------------------------------------------------- /models/Film.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/models/Film.py -------------------------------------------------------------------------------- /models/Informer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/models/Informer.py -------------------------------------------------------------------------------- /models/Linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/models/Linear.py -------------------------------------------------------------------------------- /models/PatchTST.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/models/PatchTST.py -------------------------------------------------------------------------------- /models/SparseTSF.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/models/SparseTSF.py -------------------------------------------------------------------------------- /models/Stat_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/models/Stat_models.py -------------------------------------------------------------------------------- /models/Transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/models/Transformer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | matplotlib 3 | pandas 4 | scikit-learn 5 | torch -------------------------------------------------------------------------------- /run_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/run_all.sh -------------------------------------------------------------------------------- /run_longExp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/run_longExp.py -------------------------------------------------------------------------------- /scripts/SparseTSF/linear-S/electricity.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/linear-S/electricity.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/linear-S/etth1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/linear-S/etth1.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/linear-S/etth2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/linear-S/etth2.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/linear-S/ettm1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/linear-S/ettm1.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/linear-S/ettm2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/linear-S/ettm2.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/linear-S/solar.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/linear-S/solar.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/linear-S/traffic.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/linear-S/traffic.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/linear-S/weather.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/linear-S/weather.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/linear/electricity.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/linear/electricity.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/linear/etth1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/linear/etth1.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/linear/etth2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/linear/etth2.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/linear/ettm1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/linear/ettm1.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/linear/ettm2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/linear/ettm2.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/linear/solar.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/linear/solar.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/linear/traffic.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/linear/traffic.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/linear/weather.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/linear/weather.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/mlp-S/electricity.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/mlp-S/electricity.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/mlp-S/etth1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/mlp-S/etth1.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/mlp-S/etth2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/mlp-S/etth2.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/mlp-S/ettm1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/mlp-S/ettm1.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/mlp-S/ettm2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/mlp-S/ettm2.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/mlp-S/solar.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/mlp-S/solar.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/mlp-S/traffic.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/mlp-S/traffic.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/mlp-S/weather.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/mlp-S/weather.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/mlp/electricity.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/mlp/electricity.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/mlp/etth1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/mlp/etth1.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/mlp/etth2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/mlp/etth2.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/mlp/ettm1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/mlp/ettm1.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/mlp/ettm2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/mlp/ettm2.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/mlp/solar.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/mlp/solar.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/mlp/traffic.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/mlp/traffic.sh -------------------------------------------------------------------------------- /scripts/SparseTSF/mlp/weather.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/scripts/SparseTSF/mlp/weather.sh -------------------------------------------------------------------------------- /utils/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/utils/augmentations.py -------------------------------------------------------------------------------- /utils/masking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/utils/masking.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/timefeatures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/utils/timefeatures.py -------------------------------------------------------------------------------- /utils/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lss-1138/SparseTSF/HEAD/utils/tools.py --------------------------------------------------------------------------------