Vitalitypi / MGSTGNN

MGSTGNN is a neural network model for traffic flow prediction

train steps

1、generate data

cd generate
python generate.py

2、training

python -u main.py 2>&1 | tee exps/PEMS04.log