├── LICENSE ├── README.md ├── data_provider ├── __init__.py ├── data_factory.py └── data_loader.py ├── dataset ├── ECG_data.csv └── covid.csv ├── exp ├── exp_basic.py ├── exp_main.py └── exp_stat.py ├── layers ├── AutoCorrelation.py ├── Autoformer_EncDec.py ├── Embed.py ├── SelfAttention_Family.py └── Transformer_EncDec.py ├── models ├── DLinear.py ├── FreTS.py └── NLinear.py ├── requirements.txt ├── run_longExp.py ├── utils ├── masking.py ├── metrics.py ├── timefeatures.py └── tools.py └── weight_plot.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/README.md -------------------------------------------------------------------------------- /data_provider/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /data_provider/data_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/data_provider/data_factory.py -------------------------------------------------------------------------------- /data_provider/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/data_provider/data_loader.py -------------------------------------------------------------------------------- /dataset/ECG_data.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/dataset/ECG_data.csv -------------------------------------------------------------------------------- /dataset/covid.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/dataset/covid.csv -------------------------------------------------------------------------------- /exp/exp_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/exp/exp_basic.py -------------------------------------------------------------------------------- /exp/exp_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/exp/exp_main.py -------------------------------------------------------------------------------- /exp/exp_stat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/exp/exp_stat.py -------------------------------------------------------------------------------- /layers/AutoCorrelation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/layers/AutoCorrelation.py -------------------------------------------------------------------------------- /layers/Autoformer_EncDec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/layers/Autoformer_EncDec.py -------------------------------------------------------------------------------- /layers/Embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/layers/Embed.py -------------------------------------------------------------------------------- /layers/SelfAttention_Family.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/layers/SelfAttention_Family.py -------------------------------------------------------------------------------- /layers/Transformer_EncDec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/layers/Transformer_EncDec.py -------------------------------------------------------------------------------- /models/DLinear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/models/DLinear.py -------------------------------------------------------------------------------- /models/FreTS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/models/FreTS.py -------------------------------------------------------------------------------- /models/NLinear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/models/NLinear.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | matplotlib 3 | pandas 4 | scikit-learn 5 | torch==1.9.0 -------------------------------------------------------------------------------- /run_longExp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/run_longExp.py -------------------------------------------------------------------------------- /utils/masking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/utils/masking.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/timefeatures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/utils/timefeatures.py -------------------------------------------------------------------------------- /utils/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/utils/tools.py -------------------------------------------------------------------------------- /weight_plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/aikunyi/FreTS/HEAD/weight_plot.py --------------------------------------------------------------------------------