| { | |
| "num_nodes": 207, | |
| "in_steps": 12, | |
| "out_steps": 12, | |
| "input_dim": 1, | |
| "output_dim": 1, | |
| "steps_per_day": 288, | |
| "input_embedding_dim": 24, | |
| "tod_embedding_dim": 24, | |
| "dow_embedding_dim": 0, | |
| "spatial_embedding_dim": 0, | |
| "adaptive_embedding_dim": 80, | |
| "num_heads": 4, | |
| "num_layers": 3, | |
| "dropout_a": 0.3, | |
| "use_mixed_proj": true, | |
| "model_dim": 128, | |
| "dataset": "METR-LA", | |
| "graph_mode": "learned", | |
| "lambda_hybrid": 0.5, | |
| "sparsity_k": null, | |
| "propagation_mode": "power", | |
| "temporal_mode": "mamba", | |
| "mamba_d_state": 16, | |
| "mamba_d_conv": 4, | |
| "mamba_expand": 2, | |
| "tcn_num_layers": 3, | |
| "tcn_kernel_size": 3, | |
| "tcn_dilation_base": 2, | |
| "tcn_dropout": 0.1, | |
| "use_zero_init": true | |
| } |