From 8ddf594d9b831b7b48f03130bd71760ce005357b Mon Sep 17 00:00:00 2001 From: majorli Date: Fri, 2 Dec 2022 02:12:52 +0000 Subject: [PATCH] fix graph wavenet readme issue link #I63WCW Signed-off-by: majorli --- .../graph_wavenet/pytorch/README.md | 24 +++++++++++++++---- 1 file changed, 19 insertions(+), 5 deletions(-) diff --git a/cv/traffic_forecast/graph_wavenet/pytorch/README.md b/cv/traffic_forecast/graph_wavenet/pytorch/README.md index 906eda99a..0e5547fa4 100644 --- a/cv/traffic_forecast/graph_wavenet/pytorch/README.md +++ b/cv/traffic_forecast/graph_wavenet/pytorch/README.md @@ -21,9 +21,23 @@ pip3 install -r requirements.txt ### Step 2.2: Process raw data -``` -# Create data directories -mkdir -p data/{METR-LA,PEMS-BAY} +```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 @@ -32,10 +46,10 @@ python3 generate_training_data.py --output_dir=data/METR-LA --traffic_df_filenam 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 ``` - -- Gitee