-
Notifications
You must be signed in to change notification settings - Fork 19
/
porto.json
38 lines (38 loc) · 973 Bytes
/
porto.json
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
{
"n_layers": 6,
"d_model": 256,
"attn_heads": 8,
"max_epoch": 30,
"batch_size": 64,
"grad_accmu_steps": 1,
"learning_rate": 2e-4,
"dataset": "porto",
"roadnetwork": "porto_roadmap_edge_porto_True_1_merge",
"geo_file": "porto_roadmap_edge_porto_True_1_merge_withdegree",
"rel_file": "porto_roadmap_edge_porto_True_1_merge_withdegree",
"merge": true,
"min_freq": 1,
"seq_len": 128,
"test_every": 50,
"temperature": 0.05,
"contra_loss_type": "simclr",
"classify_label": "usrid",
"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"
}