Based on Paddle implementation paper: Temporal Fusion Transformers for Interpretable Multi-horizon Time Series Forecasting
datasets:Electricity dataset
You can run it on aistudio
- link:https://pan.baidu.com/s/1XpOY04lt37NlzcEMM3XZEQ
- code:zjvu
-
Convert TXT files to CSV files
-
We have already provided a csv file dataset on aistudio so you can skip this step
python scripts/convert_data.py
-
Verifying configuration files
python config/conf.py
-
Verifying dataset
python dataset/ts_dataset.py
-
Training with single GPU
python main.py --exp_name electricity --conf_file_path your_file_path --inference False
-
Inference with best model saved
python main.py --exp_name electricity --conf_file_path your_file_path --inference True
paper | tensorflow(source code) | paddlepaddle | |
---|---|---|---|
P50 | 0.055 | 0.056 | 0.054 |
P90 | 0.027 | 0.030 | 0.028 |