Skip to main content

UCTB (Urban Computing Tool Box)

Python PyPI https://img.shields.io/badge/license-MIT-green Documentation


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, ridesharing 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 tutorial section.

News

2021-11: Our paper on UCTB, entitled 'Exploring the Generalizability of Spatio-Temporal Traffic Prediction: Meta-Modeling and an Analytic Framework', has been accepted by IEEE TKDE! [IEEE Xplore][arXiv]

2023-06: We have released a technical report entitled 'UCTB: An Urban Computing Tool Box for Spatiotemporal Crowd Flow Prediction', introducing the design and implementation principles of UCTB. [arXiv]


Urban Datasets

UCTB releases a public dataset repository including the following applications:

Application City Granularity Download Link
Bike-sharing NYC 5 minutes 66.0M
Bike-sharing Chicago 5 minutes 30.2M
Bike-sharing DC 5 minutes 31.0M
Pedestrian Count Melbourne 60 minutes 9.44M
Vehicle Speed LA 5 minutes 11.8M
Vehicle Speed BAY 5 minutes 27.9M
Ride-sharing Chicago 60 minutes 17.5M

We provide detailed documents about how to build and how to use these datasets.


Prediction Models

Currently, the package supports the following models: (This toolbox is constructed based on some open-source repos. We appreciate these awesome implements. See more details).

Model Name Input 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 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
STMeta [TKDE 2021] Node GNN Prior(Proximity,Functionality,Interaction/Same-line) LSTM/RNN Closeness,Period,Trend UCTB.model.STMeta

Visualization Tool

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!

Installation

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 or use our docker environment. The installation details are in our documents.

Release files for UCTB 0.3.5

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for UCTB 0.3.5
File Size Uploaded
UCTB-0.3.5.tar.gz 85.5 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for UCTB 0.3.5
File Interpreter ABI Platform
UCTB-0.3.5-py3-none-any.whl Python 3 none any Details

Total release size: 189.4 kB

Release files / UCTB-0.3.5.tar.gz

Download URL UCTB-0.3.5.tar.gz
Size 85.5 kB
Tags Source
SHA-256 checksum
How to use checksums
e571e372015dc618ca25f2d10e59c2cf2fe093a351d3cf1f36d6958079446f1b
BLAKE2b-256 checksum
How to use checksums
e8e06d22de791eb47f9a5e823672f0a356a08224629117eaa15fa15572eff69f
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/4.0.2 CPython/3.7.2

Release files / UCTB-0.3.5-py3-none-any.whl

Download URL UCTB-0.3.5-py3-none-any.whl
Size 103.9 kB
Tags Python 3
SHA-256 checksum
How to use checksums
20d7dc4bdb63b475216271d9aba2263f334dd4796ae51757ae5f2af2b4505eee
BLAKE2b-256 checksum
How to use checksums
70d65b92d0913ee5322d9ed997552f8d7f4b74f9ccd5c169facd343474294e60
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/4.0.2 CPython/3.7.2

Release history Release notifications | RSS feed

This release

0.3.5 This release

2 release files

0.3.0

2 release files

0.0.5

2 release files

0.0.4

2 release files

0.0.3

3 release files

0.0.2

2 release files

0.0.1

2 release files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page