Table 2 Dataset settings.
From: Spatio-temporal transformer and graph convolutional networks based traffic flow prediction
Dataset | Encoder layers | Decoder layers | Kernel size | Train: val: test |
---|---|---|---|---|
PEMS04 | 4 | 4 | 3 | 6:2:2 |
PEMS07 | 3 | 3 | 3 | 6:2:2 |
PEMS08 | 4 | 4 | 3 | 6:2:2 |
HZMetro | 3 | 3 | 3 | 18:2:5 |
SHMetro | 3 | 3 | 3 | 62:9:21 |