File size: 311 Bytes
765f977
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
{
  "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"
}