Temporal Fusion Transformer using PaddlePaddle

September 28, 2021 · View on GitHub

Based on Paddle implementation paper: Temporal Fusion Transformers for Interpretable Multi-horizon Time Series Forecasting

datasets:Electricity dataset

You can run it on aistudio

You can download the best model we have trained from Baidu Pan

The training steps:

  • 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
    

The experimental results

The Retrieval results

papertensorflow(source code)paddlepaddle
P500.0550.0560.054
P900.0270.0300.028

The reference material