spatio-temporal preidction raw code