This is the source codes of ST-Tran If you wanna test ST-Tran, run 'python train.py -c -s -FS -no-train' If you wanna test ST-Tran w/o STB in fusion, run 'python train.py -c -s -no-train'
If you wanna train the model by yourself, run 'python train.py -s -c -FS -train' or you can change the arguments as you want
The raw data can be download from https://dataverse.harvard.edu/dataverse/bigdatachallenge