Hugging Face's logo

witgaw
/
DCRNN_METR-LA

Safetensors
traffic-forecasting
time-series
graph-neural-network
dcrnn
Model card Files
xet
 Community
DCRNN_METR-LA
File size: 983 Bytes
3d0d2e7
879f02e
3d0d2e7
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
{
  "epoch": 72,
  "scaler": {
    "mean": "54.405926",
    "std": "19.49374"
  },
  "model_config": {
    "cl_decay_steps": 2000,
    "filter_type": "dual_random_walk",
    "horizon": 12,
    "input_dim": 2,
    "l1_decay": 0,
    "max_diffusion_step": 2,
    "num_nodes": 207,
    "num_rnn_layers": 2,
    "output_dim": 1,
    "rnn_units": 64,
    "seq_len": 12,
    "use_curriculum_learning": true
  },
  "data_config": {
    "batch_size": 64,
    "dataset_dir": "data/METR-LA",
    "test_batch_size": 64,
    "val_batch_size": 64,
    "graph_pkl_filename": "data/sensor_graph/adj_mx.pkl"
  },
  "train_config": {
    "base_lr": 0.01,
    "dropout": 0,
    "epoch": 0,
    "epochs": 100,
    "epsilon": 0.001,
    "global_step": 0,
    "lr_decay_ratio": 0.1,
    "max_grad_norm": 5,
    "max_to_keep": 100,
    "min_learning_rate": 2e-06,
    "optimizer": "adam",
    "patience": 50,
    "steps": [
      20,
      30,
      40,
      50
    ],
    "test_every_n_epochs": 10
  }
}