Forecasting. The forecasting module is based on the Graph Attention Network (GAT) [1]} architecture.
\(\mathbf {h}_{i}=\mathrm {ReLU}\left(\sum _{j \in \mathcal {N}(i) \cup \lbrace i\rbrace } \alpha _{ij} \mathbf {W}\mathbf {z}_{j} \right),\) 
