Code for the paper "Spatio-Temporal Neural Networks for Space-Time Series Modeling and Relations Discovery"