├── README.md ├── baselines ├── DSTMamba │ ├── DSTMamba_Arch.py │ ├── Embed.py │ ├── LTSFConfig │ │ └── PEMS08_IN96_OUT12.yaml │ ├── MambaEnc.py │ ├── RevIN.py │ ├── STFConfig │ │ └── METRLA_IN12_OUT12.yaml │ ├── SeriesDec.py │ ├── SeriesMix.py │ └── __init__.py ├── SMamba │ ├── Embed.py │ ├── LTSFConfig │ │ └── PEMS08_IN96_OUT12.yaml │ ├── Mamba_Enc.py │ ├── SMamba_Arch.py │ └── __init__.py └── __init__.py ├── checkpoints └── .gitkeep ├── data ├── data_preparation │ ├── generate_LTSF_METRLA_data.py │ └── generate_STF_METRLA_data.py ├── datasets_zoo │ └── METRLA │ │ └── .gitkeep ├── get_dataloader.py └── raw_data │ └── METRLA │ └── .gitkeep ├── exp_main.py ├── lib ├── metrics.py ├── scalers.py └── utils.py ├── logs └── .gitkeep ├── runners ├── BaseRunner.py ├── LTSFRunner.py ├── STFRunner.py └── __init__.py └── scripts ├── LTSF ├── METRLA │ └── DSTMamba.sh └── PEMS08 │ └── DSTMamba.sh └── STF └── .gitkeep /README.md: -------------------------------------------------------------------------------- 1 | Incomplete version to be improved in 2025. 2 | 3 | bash scripts/LTSF/METRLA/DSTMamba.sh 4 | -------------------------------------------------------------------------------- /baselines/DSTMamba/DSTMamba_Arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/baselines/DSTMamba/DSTMamba_Arch.py -------------------------------------------------------------------------------- /baselines/DSTMamba/Embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/baselines/DSTMamba/Embed.py -------------------------------------------------------------------------------- /baselines/DSTMamba/LTSFConfig/PEMS08_IN96_OUT12.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/baselines/DSTMamba/LTSFConfig/PEMS08_IN96_OUT12.yaml -------------------------------------------------------------------------------- /baselines/DSTMamba/MambaEnc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/baselines/DSTMamba/MambaEnc.py -------------------------------------------------------------------------------- /baselines/DSTMamba/RevIN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/baselines/DSTMamba/RevIN.py -------------------------------------------------------------------------------- /baselines/DSTMamba/STFConfig/METRLA_IN12_OUT12.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/baselines/DSTMamba/STFConfig/METRLA_IN12_OUT12.yaml -------------------------------------------------------------------------------- /baselines/DSTMamba/SeriesDec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/baselines/DSTMamba/SeriesDec.py -------------------------------------------------------------------------------- /baselines/DSTMamba/SeriesMix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/baselines/DSTMamba/SeriesMix.py -------------------------------------------------------------------------------- /baselines/DSTMamba/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/baselines/DSTMamba/__init__.py -------------------------------------------------------------------------------- /baselines/SMamba/Embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/baselines/SMamba/Embed.py -------------------------------------------------------------------------------- /baselines/SMamba/LTSFConfig/PEMS08_IN96_OUT12.yaml: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /baselines/SMamba/Mamba_Enc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/baselines/SMamba/Mamba_Enc.py -------------------------------------------------------------------------------- /baselines/SMamba/SMamba_Arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/baselines/SMamba/SMamba_Arch.py -------------------------------------------------------------------------------- /baselines/SMamba/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/baselines/SMamba/__init__.py -------------------------------------------------------------------------------- /baselines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/baselines/__init__.py -------------------------------------------------------------------------------- /checkpoints/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/data_preparation/generate_LTSF_METRLA_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/data/data_preparation/generate_LTSF_METRLA_data.py -------------------------------------------------------------------------------- /data/data_preparation/generate_STF_METRLA_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/data/data_preparation/generate_STF_METRLA_data.py -------------------------------------------------------------------------------- /data/datasets_zoo/METRLA/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/get_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/data/get_dataloader.py -------------------------------------------------------------------------------- /data/raw_data/METRLA/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /exp_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/exp_main.py -------------------------------------------------------------------------------- /lib/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/lib/metrics.py -------------------------------------------------------------------------------- /lib/scalers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/lib/scalers.py -------------------------------------------------------------------------------- /lib/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/lib/utils.py -------------------------------------------------------------------------------- /logs/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /runners/BaseRunner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/runners/BaseRunner.py -------------------------------------------------------------------------------- /runners/LTSFRunner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/runners/LTSFRunner.py -------------------------------------------------------------------------------- /runners/STFRunner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/runners/STFRunner.py -------------------------------------------------------------------------------- /runners/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/runners/__init__.py -------------------------------------------------------------------------------- /scripts/LTSF/METRLA/DSTMamba.sh: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/LTSF/PEMS08/DSTMamba.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Anle-He/DST-Mamba/HEAD/scripts/LTSF/PEMS08/DSTMamba.sh -------------------------------------------------------------------------------- /scripts/STF/.gitkeep: -------------------------------------------------------------------------------- 1 | --------------------------------------------------------------------------------