Name		Name	Last commit message	Last commit date
Latest commit History 4 Commits
data/PeMSD4		data/PeMSD4
lib		lib
README.md		README.md
model.py		model.py
run.py		run.py
train.py		train.py
utils.py		utils.py

Repository files navigation

Digital Twin Mobility Profiling: A Spatio-Temporal Graph Learning Approach

This is the implementation of traffic prediction code in DTMP based on PyTorch.

structure of the code:

data folder: storing PEMSD4 and PEMSD8 dataset. You may refer to ASTGCN for more details;
lib folder: some methods for data loading and processing from AGCRN;
utils.py: method of loading adjacency graph;
model.py: implementation of Anet;
train.py, run.py: train and run the model.

You can use python run.py --dataset PEMSD4 --num_nodes 307 command to run the code.

About

No description, website, or topics provided.

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%