forked from LiuChuang0059/Hi-GMAE
-
Notifications
You must be signed in to change notification settings - Fork 0
/
configs.yml
145 lines (145 loc) · 2.36 KB
/
configs.yml
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
IMDB-BINARY:
lr: 0.00015
lr_f: 0.005
num_hidden: 512
num_heads: 2
weight_decay: 0
weight_decay_f: 0
max_epoch: 50
max_epoch_f: 500
mask_rate: 0.3
num_layers: 1
encoder: gin
decoder: gin
activation: prelu
in_drop: 0.2
loss_fn: sce
optimizer: adam
replace_rate: 0.05
drop_edge_rate: 0.0
norm: batchnorm
residual: False
scheduler: False
linear_prob: True
pooling: mean_max
batch_size: 32
alpha_l: 1
IMDB-MULTI:
lr: 0.00015
num_hidden: 512
num_heads: 2
weight_decay: 0
max_epoch: 50
mask_rate: 0.3
num_layers: 1
encoder: gin
decoder: gin
activation: prelu
in_drop: 0.2
loss_fn: sce
optimizer: adam
replace_rate: 0.05
drop_edge_rate: 0.0
alpha_l: 1
norm: batchnorm
scheduler: False
linear_prob: True
pooling: mean
batch_size: 32
pe_dim: 10
PROTEINS:
lr: 0.00015
num_hidden: 512
weight_decay: 0
max_epoch: 100
num_layers: 3
encoder: gin
decoder: gin
activation: prelu
in_drop: 0.2
loss_fn: sce
optimizer: adam
mask_rate: 0.6
drop_edge_rate: 0.0
replace_rate: 0.05
alpha_l: 1
norm: batchnorm
scheduler: False
linear_prob: True
pooling: max
batch_size: 32
REDDIT-BINARY:
lr: 0.006
weight_decay: 0.0
num_heads: 4
max_epoch: 100
mask_rate: 0.6
drop_edge_rate: 0.0
num_hidden: 512
num_layers: 2
encoder: gcn
decoder: gcn
activation: prelu
pooling: max
scheduler: False
batch_size: 8
replace_rate: 0.1
norm: layernorm
loss_fn: sce
alpha_l: 1
COLLAB:
lr: 0.00015
weight_decay: 0.0
max_epoch: 20
num_layers: 1
num_hidden: 512
mask_rate: 0.5
replace_rate: 0.05
drop_edge_rate: 0.0
activation: relu
encoder: gin
decoder: gin
scheduler: True
pooling: max
batch_size: 32
loss_fn: sce
norm: batchnorm
alpha_l: 1
NCI1:
lr: 0.0001
max_epoch: 300
num_layers: 3
num_hidden: 512
mask_rate: 0.25
drop_edge_rate: 0.0
activation: prelu
encoder: gin
decoder: gin
scheduler: True
pool: sum
batch_size: 16
alpha_l: 2
replace_rate: 0.1
norm: batchnorm
loss_fn: sce
DD:
lr: 0.00015
num_hidden: 512
weight_decay: 0
max_epoch: 100
mask_rate: 0.3
num_layers: 1
encoder: gin
decoder: gin
activation: prelu
in_drop: 0.2
loss_fn: sce
optimizer: adam
drop_edge_rate: 0.0
alpha_l: 1
replace_rate: 0.05
norm: batchnorm
scheduler: False
linear_prob: True
pooling: max
batch_size: 32