├── README.md ├── assets └── S-Mamba.png ├── data_provider ├── __init__.py ├── data_factory.py └── data_loader.py ├── experiments ├── exp_basic.py ├── exp_long_term_forecasting.py └── exp_long_term_forecasting_partial.py ├── layers ├── AutoCorrelation.py ├── Autoformer_EncDec.py ├── Autoformer_EncDec_M.py ├── Embed.py ├── Mamba_EncDec.py ├── SelfAttention_Family.py ├── Transformer_EncDec.py ├── Transformer_EncDec_M.py └── __init__.py ├── model ├── Autoformer.py ├── Autoformer_M.py ├── Flashformer.py ├── Flashformer_M.py ├── Flowformer.py ├── Flowformer_M.py ├── Informer.py ├── Informer_M.py ├── Reformer.py ├── Reformer_M.py ├── S_Mamba.py ├── Transformer.py ├── Transformer_M.py ├── __init__.py ├── iFlashformer.py ├── iFlowformer.py ├── iInformer.py ├── iReformer.py └── iTransformer.py ├── requirements.txt ├── run.py ├── scripts ├── Reorder_ECL │ ├── S_Mamba_behind.sh │ └── S_Mamba_middle.sh ├── increasing_lookback │ ├── ECL │ │ ├── Informer.sh │ │ ├── Informer_M.sh │ │ ├── Reformer.sh │ │ ├── Reformer_M.sh │ │ ├── S_Mamba.sh │ │ ├── Transformer.sh │ │ ├── Transformer_M.sh │ │ └── iTransformer.sh │ ├── ETTm1 │ │ ├── Informer.sh │ │ ├── Informer_M.sh │ │ ├── Reformer.sh │ │ ├── Reformer_M.sh │ │ ├── S_Mamba.sh │ │ ├── Transformer.sh │ │ ├── Transformer_M.sh │ │ └── iTransformer.sh │ ├── PEMS_04 │ │ ├── Informer.sh │ │ ├── Informer_M.sh │ │ ├── Reformer.sh │ │ ├── Reformer_M.sh │ │ ├── S_Mamba.sh │ │ ├── Transformer.sh │ │ ├── Transformer_M.sh │ │ └── iTransformer.sh │ └── Traffic │ │ ├── Informer.sh │ │ ├── Informer_M.sh │ │ ├── Reformer.sh │ │ ├── Reformer_M.sh │ │ ├── S_Mamba.sh │ │ ├── Transformer.sh │ │ ├── Transformer_M.sh │ │ └── iTransformer.sh ├── mamba_as_encoder │ ├── ECL │ │ ├── Autoformer.sh │ │ ├── Autoformer_M.sh │ │ ├── Flashformer.sh │ │ ├── Flashformer_M.sh │ │ ├── Flowformer.sh │ │ └── Flowformer_M.sh │ ├── ETTm2 │ │ ├── Autoformer.sh │ │ ├── Autoformer_M.sh │ │ ├── Flashformer.sh │ │ ├── Flashformer_M.sh │ │ ├── Flowformer.sh │ │ └── Flowformer_M.sh │ ├── PEMS_07 │ │ ├── Autoformer.sh │ │ ├── Autoformer_M.sh │ │ ├── Flashformer.sh │ │ ├── Flashformer_M.sh │ │ ├── Flowformer.sh │ │ └── Flowformer_M.sh │ └── Traffic │ │ ├── Autoformer.sh │ │ ├── Autoformer_M.sh │ │ ├── Flashformer.sh │ │ ├── Flashformer_M.sh │ │ ├── Flowformer.sh │ │ └── Flowformer_M.sh ├── multivariate_forecasting │ ├── ECL │ │ └── S_Mamba.sh │ ├── ETT │ │ ├── S_Mamba_ETTh1.sh │ │ ├── S_Mamba_ETTh2.sh │ │ ├── S_Mamba_ETTm1.sh │ │ └── S_Mamba_ETTm2.sh │ ├── Exchange │ │ └── S_Mamba.sh │ ├── PEMS │ │ ├── S_Mamba_03.sh │ │ ├── S_Mamba_04.sh │ │ ├── S_Mamba_07.sh │ │ └── S_Mamba_08.sh │ ├── SolarEnergy │ │ └── S_Mamba.sh │ ├── Traffic │ │ └── S_Mamba.sh │ └── Weather │ │ └── S_Mamba.sh └── variate_generalization │ ├── ECL │ └── S_Mamba.sh │ ├── SolarEnergy │ └── S_Mamba.sh │ └── Traffic │ └── S_Mamba.sh └── utils ├── __init__.py ├── masking.py ├── metrics.py ├── timefeatures.py └── tools.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/README.md -------------------------------------------------------------------------------- /assets/S-Mamba.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/assets/S-Mamba.png -------------------------------------------------------------------------------- /data_provider/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data_provider/data_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/data_provider/data_factory.py -------------------------------------------------------------------------------- /data_provider/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/data_provider/data_loader.py -------------------------------------------------------------------------------- /experiments/exp_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/experiments/exp_basic.py -------------------------------------------------------------------------------- /experiments/exp_long_term_forecasting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/experiments/exp_long_term_forecasting.py -------------------------------------------------------------------------------- /experiments/exp_long_term_forecasting_partial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/experiments/exp_long_term_forecasting_partial.py -------------------------------------------------------------------------------- /layers/AutoCorrelation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/layers/AutoCorrelation.py -------------------------------------------------------------------------------- /layers/Autoformer_EncDec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/layers/Autoformer_EncDec.py -------------------------------------------------------------------------------- /layers/Autoformer_EncDec_M.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/layers/Autoformer_EncDec_M.py -------------------------------------------------------------------------------- /layers/Embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/layers/Embed.py -------------------------------------------------------------------------------- /layers/Mamba_EncDec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/layers/Mamba_EncDec.py -------------------------------------------------------------------------------- /layers/SelfAttention_Family.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/layers/SelfAttention_Family.py -------------------------------------------------------------------------------- /layers/Transformer_EncDec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/layers/Transformer_EncDec.py -------------------------------------------------------------------------------- /layers/Transformer_EncDec_M.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/layers/Transformer_EncDec_M.py -------------------------------------------------------------------------------- /layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/Autoformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/model/Autoformer.py -------------------------------------------------------------------------------- /model/Autoformer_M.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/model/Autoformer_M.py -------------------------------------------------------------------------------- /model/Flashformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/model/Flashformer.py -------------------------------------------------------------------------------- /model/Flashformer_M.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/model/Flashformer_M.py -------------------------------------------------------------------------------- /model/Flowformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/model/Flowformer.py -------------------------------------------------------------------------------- /model/Flowformer_M.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/model/Flowformer_M.py -------------------------------------------------------------------------------- /model/Informer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/model/Informer.py -------------------------------------------------------------------------------- /model/Informer_M.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/model/Informer_M.py -------------------------------------------------------------------------------- /model/Reformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/model/Reformer.py -------------------------------------------------------------------------------- /model/Reformer_M.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/model/Reformer_M.py -------------------------------------------------------------------------------- /model/S_Mamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/model/S_Mamba.py -------------------------------------------------------------------------------- /model/Transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/model/Transformer.py -------------------------------------------------------------------------------- /model/Transformer_M.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/model/Transformer_M.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/iFlashformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/model/iFlashformer.py -------------------------------------------------------------------------------- /model/iFlowformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/model/iFlowformer.py -------------------------------------------------------------------------------- /model/iInformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/model/iInformer.py -------------------------------------------------------------------------------- /model/iReformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/model/iReformer.py -------------------------------------------------------------------------------- /model/iTransformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/model/iTransformer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/run.py -------------------------------------------------------------------------------- /scripts/Reorder_ECL/S_Mamba_behind.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/Reorder_ECL/S_Mamba_behind.sh -------------------------------------------------------------------------------- /scripts/Reorder_ECL/S_Mamba_middle.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/Reorder_ECL/S_Mamba_middle.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/ECL/Informer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/ECL/Informer.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/ECL/Informer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/ECL/Informer_M.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/ECL/Reformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/ECL/Reformer.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/ECL/Reformer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/ECL/Reformer_M.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/ECL/S_Mamba.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/ECL/S_Mamba.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/ECL/Transformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/ECL/Transformer.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/ECL/Transformer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/ECL/Transformer_M.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/ECL/iTransformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/ECL/iTransformer.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/ETTm1/Informer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/ETTm1/Informer.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/ETTm1/Informer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/ETTm1/Informer_M.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/ETTm1/Reformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/ETTm1/Reformer.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/ETTm1/Reformer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/ETTm1/Reformer_M.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/ETTm1/S_Mamba.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/ETTm1/S_Mamba.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/ETTm1/Transformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/ETTm1/Transformer.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/ETTm1/Transformer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/ETTm1/Transformer_M.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/ETTm1/iTransformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/ETTm1/iTransformer.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/PEMS_04/Informer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/PEMS_04/Informer.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/PEMS_04/Informer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/PEMS_04/Informer_M.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/PEMS_04/Reformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/PEMS_04/Reformer.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/PEMS_04/Reformer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/PEMS_04/Reformer_M.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/PEMS_04/S_Mamba.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/PEMS_04/S_Mamba.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/PEMS_04/Transformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/PEMS_04/Transformer.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/PEMS_04/Transformer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/PEMS_04/Transformer_M.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/PEMS_04/iTransformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/PEMS_04/iTransformer.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/Traffic/Informer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/Traffic/Informer.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/Traffic/Informer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/Traffic/Informer_M.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/Traffic/Reformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/Traffic/Reformer.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/Traffic/Reformer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/Traffic/Reformer_M.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/Traffic/S_Mamba.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/Traffic/S_Mamba.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/Traffic/Transformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/Traffic/Transformer.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/Traffic/Transformer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/Traffic/Transformer_M.sh -------------------------------------------------------------------------------- /scripts/increasing_lookback/Traffic/iTransformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/increasing_lookback/Traffic/iTransformer.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/ECL/Autoformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/ECL/Autoformer.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/ECL/Autoformer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/ECL/Autoformer_M.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/ECL/Flashformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/ECL/Flashformer.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/ECL/Flashformer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/ECL/Flashformer_M.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/ECL/Flowformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/ECL/Flowformer.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/ECL/Flowformer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/ECL/Flowformer_M.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/ETTm2/Autoformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/ETTm2/Autoformer.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/ETTm2/Autoformer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/ETTm2/Autoformer_M.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/ETTm2/Flashformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/ETTm2/Flashformer.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/ETTm2/Flashformer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/ETTm2/Flashformer_M.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/ETTm2/Flowformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/ETTm2/Flowformer.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/ETTm2/Flowformer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/ETTm2/Flowformer_M.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/PEMS_07/Autoformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/PEMS_07/Autoformer.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/PEMS_07/Autoformer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/PEMS_07/Autoformer_M.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/PEMS_07/Flashformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/PEMS_07/Flashformer.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/PEMS_07/Flashformer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/PEMS_07/Flashformer_M.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/PEMS_07/Flowformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/PEMS_07/Flowformer.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/PEMS_07/Flowformer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/PEMS_07/Flowformer_M.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/Traffic/Autoformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/Traffic/Autoformer.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/Traffic/Autoformer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/Traffic/Autoformer_M.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/Traffic/Flashformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/Traffic/Flashformer.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/Traffic/Flashformer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/Traffic/Flashformer_M.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/Traffic/Flowformer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/Traffic/Flowformer.sh -------------------------------------------------------------------------------- /scripts/mamba_as_encoder/Traffic/Flowformer_M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/mamba_as_encoder/Traffic/Flowformer_M.sh -------------------------------------------------------------------------------- /scripts/multivariate_forecasting/ECL/S_Mamba.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/multivariate_forecasting/ECL/S_Mamba.sh -------------------------------------------------------------------------------- /scripts/multivariate_forecasting/ETT/S_Mamba_ETTh1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/multivariate_forecasting/ETT/S_Mamba_ETTh1.sh -------------------------------------------------------------------------------- /scripts/multivariate_forecasting/ETT/S_Mamba_ETTh2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/multivariate_forecasting/ETT/S_Mamba_ETTh2.sh -------------------------------------------------------------------------------- /scripts/multivariate_forecasting/ETT/S_Mamba_ETTm1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/multivariate_forecasting/ETT/S_Mamba_ETTm1.sh -------------------------------------------------------------------------------- /scripts/multivariate_forecasting/ETT/S_Mamba_ETTm2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/multivariate_forecasting/ETT/S_Mamba_ETTm2.sh -------------------------------------------------------------------------------- /scripts/multivariate_forecasting/Exchange/S_Mamba.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/multivariate_forecasting/Exchange/S_Mamba.sh -------------------------------------------------------------------------------- /scripts/multivariate_forecasting/PEMS/S_Mamba_03.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/multivariate_forecasting/PEMS/S_Mamba_03.sh -------------------------------------------------------------------------------- /scripts/multivariate_forecasting/PEMS/S_Mamba_04.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/multivariate_forecasting/PEMS/S_Mamba_04.sh -------------------------------------------------------------------------------- /scripts/multivariate_forecasting/PEMS/S_Mamba_07.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/multivariate_forecasting/PEMS/S_Mamba_07.sh -------------------------------------------------------------------------------- /scripts/multivariate_forecasting/PEMS/S_Mamba_08.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/multivariate_forecasting/PEMS/S_Mamba_08.sh -------------------------------------------------------------------------------- /scripts/multivariate_forecasting/SolarEnergy/S_Mamba.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/multivariate_forecasting/SolarEnergy/S_Mamba.sh -------------------------------------------------------------------------------- /scripts/multivariate_forecasting/Traffic/S_Mamba.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/multivariate_forecasting/Traffic/S_Mamba.sh -------------------------------------------------------------------------------- /scripts/multivariate_forecasting/Weather/S_Mamba.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/multivariate_forecasting/Weather/S_Mamba.sh -------------------------------------------------------------------------------- /scripts/variate_generalization/ECL/S_Mamba.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/variate_generalization/ECL/S_Mamba.sh -------------------------------------------------------------------------------- /scripts/variate_generalization/SolarEnergy/S_Mamba.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/variate_generalization/SolarEnergy/S_Mamba.sh -------------------------------------------------------------------------------- /scripts/variate_generalization/Traffic/S_Mamba.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/scripts/variate_generalization/Traffic/S_Mamba.sh -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/masking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/utils/masking.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/timefeatures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/utils/timefeatures.py -------------------------------------------------------------------------------- /utils/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wzhwzhwzh0921/S-D-Mamba/HEAD/utils/tools.py --------------------------------------------------------------------------------