34 lines
764 B
JSON
34 lines
764 B
JSON
{
|
|
"verbose": 0,
|
|
"num_rnn_layers": 2,
|
|
"epochs": 100,
|
|
"patience": 50,
|
|
"test_ratio": 0.2,
|
|
"cl_decay_steps": 2000,
|
|
"graph_pkl_filename": "data/sensor_graph/adj_mx.pkl",
|
|
"global_step": 0,
|
|
"max_diffusion_step": 2,
|
|
"epoch": 0,
|
|
"lr_decay_epoch": 20,
|
|
"learning_rate": 0.01,
|
|
"validation_ratio": 0.1,
|
|
"data_type": "ALL",
|
|
"dropout": 0.0,
|
|
"batch_size": 64,
|
|
"max_grad_norm": 5.0,
|
|
"min_learning_rate": 2e-06,
|
|
"use_cpu_only": false,
|
|
"l1_decay": 0.0,
|
|
"loss_func": "MAE",
|
|
"write_db": false,
|
|
"lr_decay": 0.1,
|
|
"lr_decay_interval": 10,
|
|
"test_every_n_epochs": 10,
|
|
"horizon": 12,
|
|
"null_val": 0.0,
|
|
"use_curriculum_learning": true,
|
|
"seq_len": 12,
|
|
"rnn_units": 64,
|
|
"base_dir": "data/model",
|
|
"filter_type": "dual_random_walk"
|
|
} |