├── .gitignore ├── LICENSE.txt ├── README.md ├── data ├── BikeNYC │ └── put_the_data_of_BikeNYC_in_this_folder.txt ├── TaxiBJ │ └── put_the_data_of_TaxiBJ_in_this_folder.txt └── TaxiNYC │ ├── Holiday.txt │ ├── Meteorology.h5 │ ├── NYC2014.h5 │ └── README.md ├── dataset ├── __init__.py ├── data_fetcher.py ├── dataset.py ├── external.py ├── minmax_normalization.py └── utils.py ├── figures └── model_arch.png ├── model ├── layer │ ├── __init__.py │ ├── concat_conv.py │ ├── conv_lstm.py │ ├── extnn.py │ ├── resnn.py │ └── seq_adaptor.py └── spn.py ├── pdf ├── ACFM_A_Dynamic_Spatial-Temporal_Network_for_Traffic_Prediction_Journal.pdf └── Attentive_Crowd_Flow_Machines_ACMMM2018.pdf ├── pretrained ├── BikeNYC │ └── model ├── TaxiBJ │ └── model └── TaxiNYC │ └── model ├── run_bikenyc.py ├── run_taxibj.py ├── run_taxinyc.py ├── test_bikenyc.py ├── test_taxibj.py ├── test_taxinyc.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/README.md -------------------------------------------------------------------------------- /data/BikeNYC/put_the_data_of_BikeNYC_in_this_folder.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/data/BikeNYC/put_the_data_of_BikeNYC_in_this_folder.txt -------------------------------------------------------------------------------- /data/TaxiBJ/put_the_data_of_TaxiBJ_in_this_folder.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/data/TaxiBJ/put_the_data_of_TaxiBJ_in_this_folder.txt -------------------------------------------------------------------------------- /data/TaxiNYC/Holiday.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/data/TaxiNYC/Holiday.txt -------------------------------------------------------------------------------- /data/TaxiNYC/Meteorology.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/data/TaxiNYC/Meteorology.h5 -------------------------------------------------------------------------------- /data/TaxiNYC/NYC2014.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/data/TaxiNYC/NYC2014.h5 -------------------------------------------------------------------------------- /data/TaxiNYC/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/data/TaxiNYC/README.md -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/data_fetcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/dataset/data_fetcher.py -------------------------------------------------------------------------------- /dataset/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/dataset/dataset.py -------------------------------------------------------------------------------- /dataset/external.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/dataset/external.py -------------------------------------------------------------------------------- /dataset/minmax_normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/dataset/minmax_normalization.py -------------------------------------------------------------------------------- /dataset/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/dataset/utils.py -------------------------------------------------------------------------------- /figures/model_arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/figures/model_arch.png -------------------------------------------------------------------------------- /model/layer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/layer/concat_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/model/layer/concat_conv.py -------------------------------------------------------------------------------- /model/layer/conv_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/model/layer/conv_lstm.py -------------------------------------------------------------------------------- /model/layer/extnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/model/layer/extnn.py -------------------------------------------------------------------------------- /model/layer/resnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/model/layer/resnn.py -------------------------------------------------------------------------------- /model/layer/seq_adaptor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/model/layer/seq_adaptor.py -------------------------------------------------------------------------------- /model/spn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/model/spn.py -------------------------------------------------------------------------------- /pdf/ACFM_A_Dynamic_Spatial-Temporal_Network_for_Traffic_Prediction_Journal.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/pdf/ACFM_A_Dynamic_Spatial-Temporal_Network_for_Traffic_Prediction_Journal.pdf -------------------------------------------------------------------------------- /pdf/Attentive_Crowd_Flow_Machines_ACMMM2018.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/pdf/Attentive_Crowd_Flow_Machines_ACMMM2018.pdf -------------------------------------------------------------------------------- /pretrained/BikeNYC/model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/pretrained/BikeNYC/model -------------------------------------------------------------------------------- /pretrained/TaxiBJ/model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/pretrained/TaxiBJ/model -------------------------------------------------------------------------------- /pretrained/TaxiNYC/model: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/pretrained/TaxiNYC/model -------------------------------------------------------------------------------- /run_bikenyc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/run_bikenyc.py -------------------------------------------------------------------------------- /run_taxibj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/run_taxibj.py -------------------------------------------------------------------------------- /run_taxinyc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/run_taxinyc.py -------------------------------------------------------------------------------- /test_bikenyc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/test_bikenyc.py -------------------------------------------------------------------------------- /test_taxibj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/test_taxibj.py -------------------------------------------------------------------------------- /test_taxinyc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/test_taxinyc.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/liulingbo918/ATFM/HEAD/utils.py --------------------------------------------------------------------------------