GraphWaveNet_PEMS-BAY / config.json
witgaw's picture
Upload Graph-WaveNet model trained on PEMS-BAY
765f977 verified
raw
history blame contribute delete
311 Bytes
{
"num_nodes": 325,
"seq_length": 12,
"horizon": 12,
"input_dim": 2,
"output_dim": 1,
"nhid": 32,
"epochs": 100,
"batch_size": 64,
"learning_rate": 0.001,
"weight_decay": 0.0001,
"dropout": 0.3,
"device": "cuda",
"gcn_bool": true,
"addaptadj": true,
"adjtype": "doubletransition"
}