995 B
995 B
Quick Run:(All based on METR-LA)
- pip install -r requirements.txt
- Download the data from https://drive.google.com/open?id=10FOTa6HXPqX8Pf5WRoRwcFnW9BrNZEIX
- Put the data into data/ for making the training data
- Create data directories
mkdir -p data/{METR-LA,PEMS-BAY}
- generate train/test/val dataset at data/{METR-LA,PEMS-BAY}/{train,val,test}.npz
python -m scripts.generate_training_data --output_dir=data/METR-LA --traffic_df_filename=data/metr-la.h5
- Constructing the Graph
python -m scripts.gen_adj_mx --sensor_ids_filename=data/sensor_graph/graph_sensor_ids.txt --normalized_k=0.1
--output_pkl_filename=data/sensor_graph/adj_mx.pkl
- Run the pre-trained model:
python run_demo_pytorch.py --config_filename=data/model/pretrained/METR-LA/config.yaml
- Model Training
python dcrnn_train_pytorch.py --config_filename=data/model/dcrnn_la.yaml
- Evaluating the baselines:
python -m scripts.eval_baseline_methods --traffic_reading_filename=data/metr-la.h5