Need help with TPA-LSTM?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

gantheory
343 Stars 109 Forks 6 Commits 13 Opened issues

Description

Temporal Pattern Attention for Multivariate Time Series Forecasting

Services available

!
?

Need anything else?

Contributors list

# 193,912
Python
6 commits

TPA-LSTM

Original Implementation of ''Temporal Pattern Attention for Multivariate Time Series Forecasting''.

Dependencies

  • python3.6.6

You can check and install other dependencies in

requirements.txt
.
$ pip install -r requirements.txt
# to install TensorFlow, you can refer to https://www.tensorflow.org/install/

Usage

The following example usage shows how to train and test a TPA-LSTM model on MuseData with settings used in this work.

Training

$ python main.py --mode train \
    --attention_len 16 \
    --batch_size 32 \
    --data_set muse \
    --dropout 0.2 \
    --learning_rate 1e-5 \
    --model_dir ./models/model \
    --num_epochs 40 \
    --num_layers 3 \
    --num_units 338

Testing

$ python main.py --mode test \
    --attention_len 16 \
    --batch_size 32 \
    --data_set muse \
    --dropout 0.2 \
    --learning_rate 1e-5 \
    --model_dir ./models/model \
    --num_epochs 40 \
    --num_layers 3 \
    --num_units 338

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.