Urban Computing Tool Box is a package providing urban datasets, spatial-temporal prediction models, and visualization tools for various urban computing tasks, such as traffic prediction, crowd flow prediction, ride-sharing demand prediction, etc.
UCTB is a flexible and open package. You can use the data we provided or use your data, and the data structure is well stated in the document.
UCTB releases a public dataset repository including the following applications in 4 scenarios, with the detailed information provided in the table below. We are constantly working to release more datasets in the future.
| Application | City | Time Span | Interval | Link |
|---|---|---|---|---|
| Bike-sharing | NYC | 2013.07.01-2017.09.30 | 5 & 60 mins | [5 mins] [60 mins] |
| Bike-sharing | Chicago | 2013.07.01-2017.09.30 | 5 & 60 mins | [5 mins] [60 mins] |
| Bike-sharing | DC | 2013.07.01-2017.09.30 | 5 & 60 mins | [5 mins] [60 mins] |
| Bus | NYC | 2022.02.01-2024.01.13 | 60 mins | [60 mins] |
| Vehicle Speed | LA | 2012.03.01-2012.06.28 | 5 mins | [5 mins] |
| Vehicle Speed | BAY | 2017.01.01-2017.07.01 | 5 mins | [5 mins] |
| Pedestrian Count | Melbourne | 2021.01.01-2022.11.01 | 60 mins | [60 mins] |
| Ride-sharing | Chicago (community) | 2013.01.01-2018.01.01 | 15 mins | [15 mins] |
| Ride-sharing | Chicago (census tract) | 2013.01.01-2018.01.01 | 15 mins | [15 mins] |
| Ride-sharing | NYC | 2009.01.01-2023.06.01 | 5 mins | [5 mins] |
| Metro | NYC | 2022.02.01-2023.12.21 | 60 mins | [60 mins] |
We provide [detailed documents] about how to use these datasets.
Currently, the ST prediction model package supports the following models: (This toolbox is constructed based on some open-source repos. We appreciate these awesome implements. [See more details]).
| Model | Data Format | Spatial Modeling Technique | Graph Type | Temporal Modeling Technique | Temporal Knowledge | Module |
|---|---|---|---|---|---|---|
| ARIMA | Both | N/A | N/A | SARIMA | Closeness | UCTB.model.ARIMA |
| HM | Both | N/A | N/A | N/A | Closeness | UCTB.model.HM |
| HMM | Both | N/A | N/A | HMM | Closeness | UCTB.model.HMM |
| XGBoost | Both | N/A | N/A | XGBoost | Closeness | UCTB.model.XGBoost |
| DeepST [SIGSPATIAL 2016] | Grid | CNN | N/A | CNN | Closeness, Period, Trend | UCTB.model.DeepST |
| ST-ResNet [AAAI 2017] | Grid | CNN | N/A | CNN | Closeness, Period, Trend | UCTB.model.ST_ResNet |
| DCRNN [ICLR 2018] | Node | GNN | Prior (Sensor Network) | RNN | Closeness | UCTB.model.DCRNN |
| GeoMAN [IJCAI 2018] | Node | Attention | Prior (Sensor Networks) | Attention+LSTM | Closeness | UCTB.model.GeoMAN |
| STGCN [IJCAI 2018] | Node | GNN | Prior (Traffic Network) | Gated CNN | Closeness | UCTB.model.STGCN |
| GraphWaveNet [IJCAI 2019] | Node | GNN | Prior (Sensor Network) + Adaptive | TCN | Closeness | UCTB.model.GraphWaveNet |
| ASTGCN [AAAI 2019] | Node | GNN+Attention | Prior (Traffic Network) | Attention | Closeness, Period, Trend | UCTB.model.ASTGCN |
| ST-MGCN [AAAI 2019] | Node | GNN | Prior (Neighborhood, Functional similarity, Transportation connectivity) | CGRNN | Closeness | UCTB.model.ST_MGCN |
| GMAN [AAAI 2020] | Node | Attention | Prior (Road Network) | Attention | Closeness | UCTB.model.GMAN |
| STSGCN [AAAI 2020] | Node | GNN+Attention | Prior (Spatial Network) | Attention | Closeness | UCTB.model.STSGCN |
| AGCRN [NeurIPS 2020] | Node | GNN | Adaptive | RNN | Closeness | UCTB.model.AGCRN |
| MTGNN [KDD 2020] | Node | GNN | Adaptive | TCN | Closeness | UCTB.model.MTGNN |
| STMeta [TKDE 2021] | Node | GNN | Prior (Proximity, Functionality, Interaction/Same-line) | LSTM/RNN | Closeness, Period, Trend | UCTB.model.STMeta |
The Visualization tool integrates visualization, error localization, and error diagnosis. Specifically, it allows data to be uploaded and provides interactive visual charts to show model errors, combined with spatiotemporal knowledge for error diagnosis.
Welcome to visit the website for a trial!
UCTB toolbox may not work successfully with the upgrade of some packages. We thus encourage you to use the specific version of packages to avoid unseen errors. To avoid potential conflict, we highly recommend you install UCTB vis Anaconda. The installation details are in our [documents].