forked from BUPT-GAMMA/OpenHGNN
-
Notifications
You must be signed in to change notification settings - Fork 0
/
best_config.py
486 lines (478 loc) · 23.6 KB
/
best_config.py
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
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
BEST_CONFIGS = {
"node_classification": {
'RGCN': {
'general': {
'seed': 0, 'patience': 40, 'validation': True, 'evaluation_metric': 'acc'
},
'aifb': {
'lr': 0.005, 'weight_decay': 0, 'max_epoch': 100,
'hidden_dim': 32, 'n_bases': -1, 'num_layers': 2, 'batch_size': 126, 'dropout': 0,
'mini_batch_flag': False, 'validation': True
},
'mutag': {
'lr': 0.005, 'weight_decay': 0.0005, 'max_epoch': 50,
'hidden_dim': 32, 'n_bases': 30, 'num_layers': 2, 'batch_size': 50, 'fanout': 4, 'dropout': 0.5,
'mini_batch_flag': False, 'validation': True
},
'bgs': {
'lr': 0.005, 'weight_decay': 0.0005, 'max_epoch': 50,
'hidden_dim': 16, 'n_bases': 40, 'num_layers': 3, 'batch_size': 126, 'fanout': 4, 'dropout': 0.1,
'mini_batch_flag': True, 'validation': True
},
'am': {
'lr': 0.01, 'weight_decay': 0.0005, 'max_epoch': 50,
'hidden_dim': 16, 'n_bases': 40, 'num_layers': 2, 'batch_size': 126, 'fanout': 4, 'dropout': 0,
'mini_batch_flag': True, 'validation': True
},
'ogbn-mag': {
'lr': 0.01, 'weight_decay': 0.0005, 'max_epoch': 10,
'hidden_dim': 16, 'n_bases': -1, 'num_layers': 2, 'batch_size': 126, 'fanout': 4, 'dropout': 0,
'mini_batch_flag': True, 'validation': True
},
'HGBn-ACM': {
'evaluation_metric': 'f1'
},
'HGBn-DBLP': {
'num_layers': 2, 'hidden_dim': 16, 'dropout': 0.5
},
'HGBn-Freebase': {
'num_layers': 5, 'hidden_dim': 16, 'dropout': 0.5
},
'HGBn-IMDB': {
},
'ohgbn-Freebase': {
'lr': 0.005, 'weight_decay': 0.0001, 'hidden_dim': 32, 'n_bases': 40, 'num_layers': 2, 'batch_size': 256,
'fanout': 4, 'dropout': 0.2, 'mini_batch_flag': True,
},
'ohgbn-acm': {
'lr': 0.01, 'hidden_dim': 64, 'dropout': 0.4, 'num_layers': 2, 'evaluation_metric': 'f1'
},
'ohgbn-imdb': {
'lr': 0.01, 'hidden_dim': 128, 'dropout': 0.2, 'num_layers': 4, 'evaluation_metric': 'f1'
},
'ohgbn-yelp2': {'lr': 0.001, 'hidden_dim': 64, 'dropout': 0.3696343130196993, 'num_layers': 3,
'use_self_loop': True, 'n_bases': 3},
'alircd_session1':{
'lr':0.001, 'hidden_dim':64, 'dropout':0.49
}
},
'CompGCN': {
'general': {
'seed': 0, 'patience': 100, 'validation': True, 'evaluation_metric': 'acc'
},
'aifb': {
'lr': 0.01, 'weight_decay': 0.0001, 'max_epoch': 100,
'hidden_dim': 32, 'num_layers': 2, 'dropout': 0.2, 'comp_fn': 'sub', 'batch_size': 128,
'mini_batch_flag': False, 'validation': True
},
'mutag': {
'lr': 0.01, 'weight_decay': 0.0001, 'max_epoch': 100,
'hidden_dim': 32, 'num_layers': 3, 'dropout': 0.5, 'comp_fn': 'sub',
'mini_batch_flag': False, 'validation': True
},
'bgs': {
},
'am': {
'lr': 0.001, 'weight_decay': 0.0001, 'max_epoch': 100, 'fanout': 4,
'hidden_dim': 32, 'num_layers': 2, 'dropout': 0.2, 'comp_fn': 'sub', 'batch_size': 128,
'mini_batch_flag': True, 'validation': True
},
'ohgbn-Freebase':{
'lr' : 0.01, 'weight_decay' : 0.0001, 'dropout' : 0.2, 'seed' : 0, 'num_layers' : 2, 'in_dim' : 32, 'hidden_dim' : 32, 'out_dim' : 32, 'max_epoch' : 500, 'patience' : 100, 'comp_fn' : 'sub', 'validation' : True, 'fanout' : 5, 'batch_size' : 128, 'mini_batch_flag' : True, 'use_uva' : True
},
'alircd_session1':{
'lr' : 0.01, 'weight_decay' : 0.0001, 'dropout' : 0.2, 'seed' : 0, 'num_layers' : 2, 'in_dim' : 32, 'hidden_dim' : 32, 'out_dim' : 32, 'max_epoch' : 500, 'patience' : 100, 'comp_fn' : 'sub', 'validation' : True, 'fanout' : 5, 'batch_size' : 128, 'mini_batch_flag' : True, 'use_uva' : True
},
'ohgbn-acm': {'lr': 0.005, 'hidden_dim': 128, 'dropout': 0.16, 'num_layers': 2,
'comp_fn': 'sub', 'evaluation_metric': 'f1'},
'ohgbn-imdb': {'lr': 0.001, 'hidden_dim': 64, 'num_layers': 3,
'comp_fn': 'sub', 'evaluation_metric': 'f1'}
},
'GTN': {
'general': {'lr': 0.005, 'weight_decay': 0.001, 'hidden_dim': 128, 'max_epoch': 100, 'patience': 20,
'norm_emd_flag': True, 'mini_batch_flag': False},
'acm4GTN': {
'num_layers': 2, 'num_channels': 2, 'adaptive_lr_flag': True,
},
'imdb4GTN': {
'lr': 0.01, 'hidden_dim': 128, 'num_layers': 2, 'num_channels': 8, 'adaptive_lr_flag': True,
'dropout': 0.1
},
'dblp4GTN': {
'num_layers': 3, 'num_channels': 2, 'adaptive_lr_flag': False,
},
'ohgbn-acm': {
'num_layers': 2, 'num_channels': 2, 'adaptive_lr_flag': True,
},
'ohgbn-imdb': {
'lr': 0.001, 'num_layers': 2, 'num_channels': 2, 'adaptive_lr_flag': True,
}
},
'fastGTN': {
'general': {'lr': 0.005, 'weight_decay': 0.001, 'hidden_dim': 64, 'max_epoch': 100, 'patience': 20,
'norm_emd_flag': True, 'mini_batch_flag': False},
'acm4GTN': {
'lr': 0.01, 'hidden_dim': 32, 'num_layers': 3, 'num_channels': 8, 'adaptive_lr_flag': False,
'dropout': 0.01,
},
'imdb4GTN': {
'lr': 0.01, 'num_layers': 3, 'dropout': 0.3, 'num_channels': 4, 'hidden_dim': 128,
'adaptive_lr_flag': False,
},
'dblp4GTN': {'lr': 0.001, 'hidden_dim': 128, 'num_layers': 3, 'num_channels': 8, 'adaptive_lr_flag': False,
'dropout': 0.5},
},
'MHNF': {
'general': {'lr': 0.005, 'weight_decay': 0.001, 'hidden_dim': 64, 'max_epoch': 50, 'patience': 10,
'norm_emd_flag': True, 'mini_batch_flag': False},
'acm4GTN': {
'num_layers': 2, 'num_channels': 2, 'adaptive_lr_flag': True,
},
'imdb4GTN': {
'lr': 0.001, 'hidden_dim': 128, 'num_layers': 4, 'num_channels': 4
},
'dblp4GTN': {
'num_layers': 3, 'num_channels': 2, 'adaptive_lr_flag': False,
},
'ohgbn-acm': {
'lr': 0.005, 'hidden_dim': 64, 'weight_decay': 0.005, 'num_layers': 2, 'num_channels': 3,
'norm_emd_flag': False, 'identity': False, 'adaptive_lr_flag': True,
},
'ohgbn-imdb': {
'lr': 0.1, 'hidden_dim': 32, 'weight_decay': 0.01, 'num_layers': 4, 'num_channels': 2,
'identity': False, 'adaptive_lr_flag': True,
},
},
'HAN': {
'general': {},
'acm_han_raw': {
'lr': 0.005, 'num_heads': [8], 'hidden_dim': 128, 'dropout': 0.6, 'weight_decay': 0.001,
'num_epochs': 200, 'patience': 100
},
'alircd_session1': {
'lr': 0.005, 'num_heads': [8], 'hidden_dim': 128, 'dropout': 0.6, 'weight_decay': 0.001,
'num_epochs': 50, 'patience': 10, 'mini_batch_flag': True, 'batch_size': 256, 'fanout': 10
},
'acm4GTN': {
'lr': 0.01, 'hidden_dim': 128, 'dropout': 0.46, 'num_heads': [1]
},
'imdb4GTN': {
'lr': 0.01, 'hidden_dim': 64, 'dropout': 0.24, 'num_heads': [2, 2]
},
'HGBn-ACM': {
},
'ohgbn-acm': {'num_heads': [8, 8]},
'ohgbn-imdb': {'lr': 0.005, 'hidden_dim': 32, 'dropout': 0.5}
},
'HPN': {
'general': {},
'ohgbn-acm': {
'lr': 0.01, 'hidden_dim': 64, 'dropout': 0.2, 'k_layer': 3, 'alpha': 0.2, 'edge_drop': 0
},
'ohgbn-imdb': {
'lr': 0.005, 'hidden_dim': 128, 'dropout': 0.25
}
},
'NARS': {
'general': {},
'HGBn-ACM': {
'lr': 0.001, 'num_heads': [8], 'num_hops': 3, 'hidden_dim': 64, 'dropout': 0.5, 'weight_decay': 0.001,
'num_epochs': 200, 'patience': 100, 'evaluation_metric': 'acc', 'ff_layer': 2,
},
'HGBn-DBLP': {
'lr': 0.001, 'num_heads': [8], 'num_hops': 3, 'hidden_dim': 512, 'dropout': 0.5, 'weight_decay': 0.001,
'num_epochs': 200, 'patience': 100, 'evaluation_metric': 'acc', 'ff_layer': 2,
},
'HGBn-Freebase': {
'lr': 0.001, 'num_heads': [8], 'num_hops': 2, 'hidden_dim': 64, 'dropout': 0.5, 'weight_decay': 0.001,
'num_epochs': 200, 'patience': 100, 'evaluation_metric': 'acc', 'ff_layer': 2,
},
'HGBn-IMDB': {
'lr': 0.001, 'num_heads': [8], 'num_hops': 2, 'hidden_dim': 64, 'dropout': 0.5, 'weight_decay': 0.001,
'num_epochs': 200, 'patience': 100, 'evaluation_metric': 'acc', 'ff_layer': 2,
},
'ohgbn-acm': {'lr': 0.001, 'hidden_dim': 32, 'dropout': 0.4, 'num_heads': 2, 'num_hops': 2,
'evaluation_metric': 'f1'},
'ohgbn-imdb': {'lr': 0.01, 'hidden_dim': 128, 'dropout': 0.4, 'num_heads': 1, 'num_hops': 2,
'evaluation_metric': 'f1'}
},
'HetGNN': {
'general': {'max_epoch': 500, 'patience': 20, 'mini_batch_flag': True},
'academic4HetGNN': {
'lr': 0.001, 'weight_decay': 0.00001, 'dim': 128, 'batch_size': 64, 'window_size': 5,
'batches_per_epoch': 50, 'rw_length': 50, 'rw_walks': 10, 'rwr_prob': 0.5,
}
},
'MAGNN': {
'general': {'lr': 0.005, 'weight_decay': 0.001, 'dropout': 0.5,
'hidden_dim': 64, 'encoder_type': 'RotateE', 'inter_attn_feats': 128, 'num_heads': 8,
'num_layers': 2},
'imdb4MAGNN': {
'max_epoch': 100, 'patience': 30, 'out_dim': 3, 'mini_batch_flag': False
},
'dblp4MAGNN': {
'max_epoch': 10, 'patience': 5, 'mini_batch_flag': True, 'batch_size': 8, 'num_samples': 5,
'num_workers': 24, 'out_dim': 4
}
},
'NSHE': {
'general': {},
'acm4NSHE': {'weight_decay': 0.001, 'num_e_neg': 1, 'num_ns_neg': 4,
'max_epoch': 500, 'patience': 10,
}
},
'HeCo': {
'general': {
'hidden_dim': 64, 'max_epoch': 10000, 'mini_batch_flag': False
},
'acm4HeCo': {'seed': 84, 'eva_lr': 0.05, 'eva_wd': 0, 'patience': 5, 'learning_rate': 0.0008,
'weight_decay': 0, 'tau': 0.8, 'feat_drop': 0.3, 'attn_drop': 0.5,
'sample_rate': {'author': 7, "subject": 3}, 'lam': 0.5
}
},
'DMGI': {
'general': {
'seed': 0, 'sc': 3, 'learning_rate': 0.0005
},
'acm_han_raw': {'weight_decay': 0.0001, 'dropout': 0.5, 'reg_coef': 0.001,
'patience': 20, 'sup_coef': 0.1, 'hid_unit': 64,
'isSemi': False, 'isBias': False, 'isAttn': False
},
'imdb4GTN': {'weight_decay': 0.0001, 'dropout': 0.5, 'reg_coef': 0.1,
'patience': 50, 'sup_coef': 0.1, 'hid_unit': 64,
'isSemi': False, 'isBias': False, 'isAttn': False
}
},
'RSHN': {
'general': {},
'aifb': {
'in_dim': 16, 'hidden_dim': 16, 'num_node_layer': 2,
'num_edge_layer': 2, 'rw_len': 4, 'batch_size': 5000,
},
'mutag': {
'hidden_dim': 16, 'num_node_layer': 2, 'num_edge_layer': 2, 'rw_len': 4, 'batch_size': 5000,
'dropout': 0.2,
},
'bgs': {
'hidden_dim': 16, 'num_node_layer': 2, 'num_edge_layer': 2, 'rw_len': 4, 'batch_size': 5000
},
'am': {'hidden_dim': 16, 'num_node_layer': 2, 'num_edge_layer': 2, 'rw_len': 4, 'batch_size': 5000
},
'ohgbn-acm': {'lr': 0.001, 'hidden_dim': 128, 'dropout': 0.4, 'num_channels': 2, 'num_node_layer': 2,
'num_edge_layer': 2},
'ohgbn-imdb': {'lr': 0.01, 'hidden_dim': 128, 'dropout': 0.09, 'num_node_layer': 2, 'num_edge_layer': 2
}
},
'HGSL': {
'general': {},
'acm4GTN': {
'undirected_relations': 'author-paper,paper-subject', 'gnn_dropout': 0, 'fs_eps': 0.8,
'fp_eps': 0.2, 'mp_eps': 0.6,'hidden_dim': 128,'num_heads': 3,'gnn_emd_dim':128,'lr':0.001,
'weight_decay':0.0001,'mini_batch_flag':False,'max_epoch':200,
}
},
'HGT': {
'general': {},
'ohgbn-acm': {
'lr': 0.001, 'hidden_dim': 128, 'dropout': 0.1, 'num_layers': 2, 'num_heads': 1
},
'ohgbn-imdb': {'lr': 0.001, 'hidden_dim': 128, 'dropout': 0.3, 'num_layers': 2, 'num_heads': 4},
'ohgbn-Freebase':{'lr':0.001, 'hidden_dim':128, 'dropout': 0.01137822722670511, 'num_heads': 2 }
},
'SimpleHGN': {
'general': {},
'ohgbn-acm': {
'lr': 0.0005, 'hidden_dim': 64, 'num_layers': 3, 'edge_dim': 32, 'num_heads': 2, 'num_edge': 4,
'attn_drop_rate': 0.1, 'feats_drop_rate': 0.5, 'beta': 0.05, 'residual': False, 'slope': 0.01,
'max_epoch': 200, 'patience': 50
},
'ohgbn-imdb': {'lr': 0.001, 'hidden_dim': 32, 'num_layers': 2, 'edge_dim': 32, 'num_heads': 8,
'num_edge': 8,
'attn_drop_rate': 0.34, 'feats_drop_rate': 0.4, 'beta': 0.01,
'residual': False, 'slope': 0.1, 'max_epoch': 100, 'patience': 50},
'ohgbn-Freebase': {
'lr':0.001, 'hidden_dim': 128, 'dropout': 0.16263095198267938, 'num_heads': 4, 'feats_drop_rate': 0.13286688066111216, 'edge_dim': 32
},
},
'RGAT':{
'general':{
'in_dim' : 64, 'num_layers' : 1, 'hidden_dim' : 64, 'patience' : 100, 'seed' : 0, 'lr' : 0.01, 'weight_decay' : 5e-4, 'max_epoch' : 350, 'dropout' : 0.2, 'num_heads' : 3,'out_dim' : 3
},
'ohgbn-Freebase':{'fanout' : 5, 'batch_size' : 128, 'mini_batch_flag' : True, 'use_uva' : True},
'alircd_session1':{'fanout' : 5, 'batch_size' : 128, 'mini_batch_flag' : True, 'use_uva' : True}
},
'RSage':{
'general':{'in_dim' : 64, 'num_layers' : 3, 'hidden_dim' : 64, 'patience' : 100, 'seed' : 0, 'lr' : 0.01, 'weight_decay' : 5e-4, 'max_epoch' : 350, 'dropout' : 0.2, 'aggregator_type' : 'pool', 'out_dim' : 3},
'alircd_session1':{'fanout' : 5, 'batch_size' : 128, 'mini_batch_flag' : True, 'use_uva' : True}
},
'HetSANN':{
'general':{'lr' : 0.0001, 'weight_decay' : 0.0005, 'dropout' : 0.2, 'seed' : 0, 'hidden_dim' : 64, 'num_layers' : 2, 'num_heads' : 16, 'max_epoch' : 10000, 'patience' : 100, 'slope' : 0.2, 'residual' : True, 'mini_batch_flag' : True, 'batch_size' : 2048, 'fanout' : 5, 'use_uva' : True},
'ohgbn-Freebase':{'lr': 0.001, 'hidden_dim': 128, 'dropout': 0.33, 'num_heads': 8},
'alircd_session1':{'lr': 0.01, 'hidden_dim': 32, 'dropout': 0.19},
},
'ieHGCN':{
'general':{'num_layers' : 5, 'hidden_dim' : 64, 'attn_dim' : 32, 'out_dim' : 16, 'patience' : 100, 'seed' : 0, 'lr' : 0.001, 'weight_decay' : 5e-4, 'max_epoch' : 3500, 'mini_batch_flag' : True, 'fanout' : 10, 'batch_size' : 512, 'dropout' : 0.2, 'bias' : True, 'batchnorm' : True},
'alircd_session1':{'lr': 0.01, 'hidden_dim': 64, 'dropout': 0.12}
}
},
"link_prediction": {
'NARS': {
'general': {'num_hops': 3},
},
'HetGNN': {
'general': {'max_epoch': 500, 'patience': 10, 'mini_batch_flag': True},
'academic4HetGNN': {
'lr': 0.01, 'weight_decay': 0.0001, 'dim': 128, 'batch_size': 64, 'window_size': 5,
'batches_per_epoch': 50, 'rw_length': 50, 'rw_walks': 10, 'rwr_prob': 0.5,
}
},
'HAN': {
'general': {
},
'HGBl-PubMed': {
'lr': 0.01, 'weight_decay': 0.0005, 'max_epoch': 400,
'hidden_dim': 64, 'n_bases': 40, 'num_layers': 4, 'batch_size': 126, 'fanout': 4, 'dropout': 0.5,
'validation': True, 'feat': 2, 'seed': 4,
},
},
'CompGCN': {
'general': {
},
'ohgbl-yelp1': {
'mini_batch_flag': True, 'num_layers': 2,
},
'ohgbl-yelp2': {
'mini_batch_flag': True, 'num_layers': 2, 'batch_size': 1024
},
'ohgbl-MTWM': {
'weight_decay': 0.00001, 'max_epoch': 50, 'patience': 50,
'mini_batch_flag': True, 'num_layers': 2,
}
},
'RGCN': {
'general': {
},
'HGBl-amazon': {
'lr': 0.01, 'weight_decay': 0.0005, 'max_epoch': 400,
'hidden_dim': 64, 'n_bases': 40, 'num_layers': 4, 'batch_size': 126, 'fanout': 4, 'dropout': 0.5,
'validation': True, 'feat': 2, 'seed': 0,
},
'HGBl-PubMed': {
'lr': 0.01, 'weight_decay': 0.0005, 'max_epoch': 400,
'hidden_dim': 64, 'n_bases': 40, 'num_layers': 4, 'batch_size': 126, 'fanout': 4, 'dropout': 0.5,
'validation': True, 'feat': 2, 'seed': 4,
},
'HGBl-LastFM': {
'lr': 0.01, 'weight_decay': 0.0005, 'max_epoch': 400,
'hidden_dim': 64, 'n_bases': 40, 'num_layers': 4, 'batch_size': 126, 'fanout': 4, 'dropout': 0.5,
'validation': True, 'feat': 2, 'seed': 3,
},
'FB15k-237': {
'lr': 0.01, 'weight_decay': 0.0005, 'max_epoch': 100,
'hidden_dim': 16, 'n_bases': 40, 'num_layers': 2, 'batch_size': 126, 'fanout': 4, 'dropout': 0,
'validation': True
},
'ohgbl-MTWM': {
'mini_batch_flag': True, 'num_layers': 2, 'batch_size': 12800
},
'ohgbl-yelp1': {
'mini_batch_flag': True, 'num_layers': 2, 'batch_size': 102400
},
'ohgbl-yelp2': {
'mini_batch_flag': True, 'num_layers': 2,
}
},
'TransE': {
'general':{
},
'FB15k':{
'lr':1, 'weight_decay': 0.0001, 'hidden_dim': 400, 'neg_size': 13, 'margin': 4, 'batch_size': 100, 'valid_percent':0.01, 'test_percent': 0.1
},
'wn18':{
# 'lr': 0.71, 'weight_decay': 0.0001, 'hidden_dim': 400, 'neg_size': 97, 'margin': 32.8, 'batch_size': 100, 'dis_norm': 1, 'valid_percent':1, 'test_percent': 1
'lr': 1, 'weight_decay': 0.0001, 'hidden_dim': 400, 'neg_size': 100, 'margin': 50, 'batch_size': 100, 'dis_norm': 2, 'patience': 3, 'valid_percent':0.05, 'test_percent': 0.05
}
},
'TransH': {
'general':{
},
'FB15k':{
'lr': 1, 'weight_decay': 0.0001, 'hidden_dim': 400, 'neg_size': 15, 'margin': 5, 'batch_size': 100, 'valid_percent':0.01, 'test_percent': 0.1
},
'wn18':{
'lr': 0.71, 'weight_decay': 0.0001, 'hidden_dim': 200, 'neg_size': 100, 'margin': 32.8, 'batch_size': 100, 'patience': 5, 'valid_percent':1, 'test_percent': 1
}
},
'TransD': {
'general':{
},
'FB15k':{
'lr': 0.6, 'weight_decay': 0.0001, 'ent_dim': 400, 'rel_dim': 400, 'neg_size': 20, 'margin': 5, 'batch_size': 100, 'patience': 3, 'valid_percent':0.01, 'test_percent': 0.1,
},
'wn18':{
'lr': 0.5, 'weight_decay': 0.0001, 'ent_dim': 400, 'rel_dim': 400, 'neg_size': 98, 'margin': 200, 'batch_size': 100, 'patience':5, 'valid_percent':1, 'test_percent': 1
}
},
'RedGNN': {
'general': {
},
'WN18RR_v1':{
'batch_size':100, 'hidden_dim' : 64, 'lr': 0.005, 'weight_decay': 0.0002, 'decay_rate': 0.991, 'attn_dim':5,
'act': 'idd', 'n_layer': 5
},
'fb237_v1': {
'batch_size':20, 'hidden_dim' : 32, 'lr': 0.0092, 'weight_decay': 0.0003, 'decay_rate': 0.994, 'attn_dim':5,
'act': 'relu', 'n_layer': 3
},
'nell_v1': {
'batch_size':10, 'hidden_dim' : 48, 'lr': 0.0021, 'weight_decay': 0.000189, 'decay_rate': 0.9937, 'attn_dim':5,
'act': 'relu', 'n_layer': 5
},
'WN18RR_v2':{
'batch_size':20, 'hidden_dim' : 48, 'lr': 0.0016, 'weight_decay': 0.0004, 'decay_rate': 0.994, 'attn_dim':3,
'act': 'relu', 'n_layer': 5
},
'fb237_v2': {
'batch_size':10, 'hidden_dim' : 48, 'lr': 0.0077, 'weight_decay': 0.0002, 'decay_rate': 0.993, 'attn_dim':5,
'act': 'relu', 'n_layer': 3
},
'nell_v2': {
'batch_size':100, 'hidden_dim' : 48, 'lr': 0.0075, 'weight_decay': 0.000189, 'decay_rate': 0.9996, 'attn_dim':5,
'act': 'relu', 'n_layer': 3
},
'WN18RR_v3':{
'batch_size':20, 'hidden_dim' : 64, 'lr': 0.0014, 'weight_decay': 0.0004, 'decay_rate': 0.994, 'attn_dim':5,
'act': 'tanh', 'n_layer': 5
},
'fb237_v3': {
'batch_size':30, 'hidden_dim' : 48, 'lr': 0.0006, 'weight_decay': 0.00023, 'decay_rate': 0.993, 'attn_dim':5,
'act': 'relu', 'n_layer': 3
},
'nell_v3': {
'batch_size':10, 'hidden_dim' : 16, 'lr': 0.0008, 'weight_decay': 0.0004, 'decay_rate': 0.995, 'attn_dim':3,
'act': 'relu', 'n_layer': 3
},
'WN18RR_v4':{
'batch_size':10, 'hidden_dim' : 32, 'lr': 0.006, 'weight_decay': 0.000132, 'decay_rate': 0.991, 'attn_dim':5,
'act': 'relu', 'n_layer': 5
},
'fb237_v4': {
'batch_size':20, 'hidden_dim' : 48, 'lr': 0.0052, 'weight_decay': 0.000018, 'decay_rate': 0.999, 'attn_dim':5,
'act': 'idd', 'n_layer': 5
},
'nell_v4': {
'batch_size':20, 'hidden_dim' : 16, 'lr': 0.0005, 'weight_decay': 0.000398, 'decay_rate': 1, 'attn_dim':3,
'act': 'tanh', 'n_layer': 5
},
}
},
"recommendation": {
'KGCN': {
"general": {},
'LastFM4KGCN': {
'in_dim': 16, 'hidden_dim': 16, 'n_relation': 60,
'batch_size': 128, 'lr': 0.002
},
}
},
}