chuanting / STDenseNet

Source code of Citywide Cellular Traffic Prediction Based on Densely Connected Convolutional Neural Networks

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Basic operation

This is the code for "Citywide Cellular Traffic Prediction Based on Densely Connected Convolutional Neural Networks".

Users can run "demo_densenet.py" to train the model and the results will be generated and stored in the reporsitory.

If you just want to see the prediction reseults without time to run the code, you can simply run "run_prediction.py", and you will also obtain the results.

Data

The data can be downloaded at my web.

The paper can be found at this link.

An executable code can also be found at codeocean.

Update 2019/06/03

The code is written by using PyTorch 0.3 and for PyTorch version > 1.0, it can not be executed successfully. I update the code and if there still exists bugs, please email me directly.

About

Source code of Citywide Cellular Traffic Prediction Based on Densely Connected Convolutional Neural Networks


Languages

Language:Python 100.0%