Skip to main content
. 2021 Jul 9;16(7):e0254319. doi: 10.1371/journal.pone.0254319

Table 2. Neural network configurations: The number of GRU layers nh, the number of ResBlocks nR, the number of TCBlocks nTC, the kernel size k, the number of kernel filters nk, and the size of key nkey in SNAIL.

For TCN and SNAIL, the value of whist is omitted as setting whist = 128 yields the best results. For TCN, nk = 4.

Seq2Seq TCN SNAIL
RMSE (nh, whist) = (3, 128) (nR, k) = (8, 4) (nTC, k, nk, nkey) = (7, 4, 4, 32)
MAPE (nh, whist) = (4, 64) (nR, k) = (8, 2) (nTC, k, nk, nkey) = (6, 2, 8, 32)
L2E (nh, whist) = (3, 128) (nR, k) = (8, 4) (nTC, k, nk, nkey) = (7, 4, 4, 32)
PCORR (nh, whist) = (5, 128) (nR, k) = (7, 4) (nTC, k, nk, nkey) = (7, 4, 4, 32)