A Spatial-temporal Attention Approach for Traffic Prediction

This is a TensorFlow1.0 implementation of APTN.

Requirements

Dependency can be installed using the following command:

pip install -r requirements.txt

# PeMSD4
python train.py --config_filename=configs/pems04.yaml

# PeMSD8
python train.py --config_filename=configs/pems08.yaml

# PeMSD4
python eval.py --config_filename={saved_model_config_filename}

# PeMSD8
python eval.py --config_filename={saved_model_config_filename}