DeepSTN+

Keras implementation of DeepSTN+: Context-aware Spatial-Temporal Neural Network for Crowd Flow Prediction in Metropolis. Ziqian Lin^, Jie Feng^, Ziyang Lu, Yong Li, and Depeng Jin. AAAI 2019. (^ indicates equal contribution) PDF If our codes is helpful to your research, you can cite our work by:

@article{lin2019deepstn+:,
title={DeepSTN+: Context-aware Spatial Temporal Neural Network for Crowd Flow Prediction in Metropolis},
author={Lin, Ziqian and Feng, Jie and Lu, Ziyang and Li, Yong and Jin, Depeng},
booktitle={Thirty-Thrid AAAI Conference on Artificial Intelligence},
year={2019}
}

Datasets

Similar to ST-ResNet, our dataset is from the NYC Bike. Besides, we collect 9 types of PoIs for this dataset. The spatial map size of the dataset is 21x12. The dataset is in the folder /DATA/dataBikeNYC flow_data.npy ( TimeLenth x In&OutFlow x MapHeight x MapWidth = 4392 x 2 x 21 x 12 ) and poi_data.npy ( PoICategories x MapHeight x MapWidth = 9 x 21 x 12 ) for directly used.

Requirements

python 3.5
Keras 2.0
NumPy

Project Structure

File BikeNYC corresponds the Dataset BikeNYC in the Paper DeepSTN+.

/DATA
- dataBikeNYC contain dataset flow_data.npy and poi_data.npy
- lzq_read_data_time_poi.py transfer flow_data.npy and poi_data.npy to the input of the DeepSTN+ network
/DST_network baseline from ST-ResNet
- ilayer.py
- metrics.py
- STResNet.py
/DeepSTN_00/SCORE are used to save the results of DeepSTN
/DeepSTN_10/SCORE are used to save the results of DeepSTN+plus
/DeepSTN_01/SCORE are used to save the results of DeepSTN+PoI$*$time
/DeepSTN_11/SCORE are used to save the results of DeepSTN+plus+PoI$*$time
/DeepSTN_network
- DeepSTN_net.py model codes for DeepSTN+
- metrics.py contains the metric RMSE
/ComparisonBikeNYC.py you can run this file to get the results of ST-ResNet and DeepSTN in the paper.

Usage

python ComparisonBikeNYC.py

Other parameters:

Refer to ComparisonBikeNYC.py and DeepSTN_net.py

for training:
- epoch, batch_size, lr, days_test, iterate_num
- XDST, X11, X10, X01, X00, trainDST, train11, train10, train01, train00
model definition: H, W, channel, T, len_closeness, len_period, len_trend, T_closeness, T_period, T_trend, pre_F, conv_F, R_N, drop, is_plus, plus, rate, is_pt, P_N, T_F, PT_F, is_summary, kernel1, isPT_F

FIBLAB / DeepSTN