Synthetic Datasets for Machine Learning on Spatio-Temporal Graphs using PDEs