cv/traffic_forecast/graph_wavenet/pytorch/README.md · DeepSpark/DeepSparkHub

# Graph WaveNet for Deep Spatial-Temporal Graph Modeling

## Model description

Spatial-temporal graph modeling is an important task to analyze the spatial relations and temporal trends of components in a system. Existing approaches mostly capture the spatial dependency on a fixed graph structure, assuming that the underlying relation between entities is pre-determined. However, the explicit graph structure (relation) does not necessarily reflect the true dependency and genuine relation may be missing due to the incomplete connections in the data. Furthermore, existing methods are ineffective to capture the temporal trends as the RNNs or CNNs employed in these methods cannot capture long-range temporal sequences. To overcome these limitations, we propose in this paper a novel graph neural network architecture, Graph WaveNet, for spatial-temporal graph modeling. By developing a novel adaptive dependency matrix and learn it through node embedding, our model can precisely capture the hidden spatial dependency in the data. With a stacked dilated 1D convolution component whose receptive field grows exponentially as the number of layers increases, Graph WaveNet is able to handle very long sequences. These two components are integrated seamlessly in a unified framework and the whole framework is learned in an end-to-end manner. Experimental results on two public traffic network datasets, METR-LA and PEMS-BAY, demonstrate the superior performance of our algorithm.

## Step 1: Installing packages

```shell
pip3 install -r requirements.txt
```

## Step 2: Preparing datasets

### Step 2.1: Download METR-LA and PEMS-BAY data from [Google Drive](https://drive.google.com/open?id=10FOTa6HXPqX8Pf5WRoRwcFnW9BrNZEIX) or [Baidu Yun](https://pan.baidu.com/s/14Yy9isAIZYdU__OYEQGa_g) links provided by [DCRNN](https://github.com/liyaguang/DCRNN).

### Step 2.2: Process raw data

```shell
# Create data and garage directories
mkdir -p data/
mkdir -p garage/

# Get adj_mx_bay.pkl and adj_mx.pkl
mkdir -p data/sensor_graph/
wget https://github.com/liyaguang/DCRNN/raw/master/data/sensor_graph/adj_mx_bay.pkl -P data/sensor_graph/
wget https://github.com/liyaguang/DCRNN/raw/master/data/sensor_graph/adj_mx.pkl -P data/sensor_graph/

# Put METR-LA and PEMS-BAY data in data/
data/
├── metr-la.h5
├── pems-bay.h5
└── sensor_graph
    ├── adj_mx_bay.pkl
    └── adj_mx.pkl

# METR-LA
python3 generate_training_data.py --output_dir=data/METR-LA --traffic_df_filename=data/metr-la.h5

# PEMS-BAY
python3 generate_training_data.py --output_dir=data/PEMS-BAY --traffic_df_filename=data/pems-bay.h5

```

## Step 3: Training

```shell
python3 train.py --gcn_bool --adjtype doubletransition --addaptadj  --randomadj
```

DeepSpark/DeepSparkHub

简介

发行版 (8)

贡献者

近期动态

DeepSpark/DeepSparkHub .gitee-modal { width: 500px !important; }

简介

发行版 (8)

开源评估指数源自 OSS-Compass 评估体系，评估体系围绕以下三个维度对项目展开评估：

贡献者

近期动态

搜索帮助

DeepSpark/DeepSparkHub