bj.json

{
  "n_layers": 6,
  "d_model": 256,
  "attn_heads": 8,
  "max_epoch": 30,
  "batch_size": 64,
  "grad_accmu_steps": 1,
  "learning_rate": 2e-4,
  "dataset": "bj",
  "roadnetwork": "bj_roadmap_edge_bj_True_1_merge",
  "geo_file": "bj_roadmap_edge_bj_True_1_merge_withdegree",
  "rel_file": "bj_roadmap_edge_bj_True_1_merge_withdegree",
  "merge": true,
  "min_freq": 1,
  "seq_len": 128,
  "test_every": 50,
  "temperature": 0.05,
  "contra_loss_type": "simclr",
  "classify_label": "vflag",
  "type_ln": "post",
  "add_cls": true,
  "add_time_in_day": true,
  "add_day_in_week": true,
  "add_pe": true,
  "add_temporal_bias": true,
  "temporal_bias_dim": 64,
  "use_mins_interval": false,
  "add_gat": true,
  "gat_heads_per_layer": [8, 16, 1],
  "gat_features_per_layer": [16, 16, 256],
  "gat_dropout": 0.1,
  "gat_K": 1,
  "gat_avg_last": true,
  "load_trans_prob": true,
  "append_degree2gcn": true,
  "normal_feature": false,
  "pooling": "cls"
}