-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathhyparams.py
151 lines (149 loc) · 13.5 KB
/
hyparams.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
hparams = {
'arxiv-year': {
'mlp': {'num_layers': 3, 'hidden_channels': 256,},
'lp': [{'hops': 1, 'alpha': 0.01}, {'hops': 2, 'alpha': 0.01}],
'link': {'weight_decay': 0.01},
'sgc': [{'hops': 1, 'weight_decay': 0.01}, {'hops': 2, 'weight_decay': 0.1}],
'cs': [{'hops': 1, 'A1': 'AD', 'A2': 'DA', 'alpha1': 0.47775554584477997, 'alpha2': 0.9292002093878098}, {'hops': 2, 'A1': 'DAD', 'A2': 'DA', 'alpha1': 0.08348704842280531, 'alpha2': 0.6555474944185927}],
# 'gcn': {'num_layers': 2, 'hidden_channels': 64, 'lr': 0.01, 'delta':1.2},
'gcn': {'num_layers': 2, 'hidden_channels': 64, 'lr': 0.01, },
'gat': {'num_layers': 2, 'hidden_channels': 32, 'gat_heads': 8, 'lr': 0.01},
'gcnjk': {'num_layers': 2, 'hidden_channels': 64, 'jk_type': 'max', 'lr': 0.01},
'gatjk': {'num_layers': 2, 'hidden_channels': 32, 'gat_heads': 8, 'jk_type': 'max', 'lr': 0.01},
'appnp': {'hidden_channels': 256, 'gpr_alpha': 0.9, 'lr': 0.01},
'h2gcn': {'num_layers': 2, 'hidden_channels': 64, 'dropout': 0.0},
'mixhop': {'num_layers': 3, 'hidden_channels': 32, 'hops': 2},
'gprgnn': {'hidden_channels': 256, 'gpr_alpha': 0.9, 'lr': 0.01},
'gcn2': {'num_layers': 8, 'gcn2_alpha': 0.5, 'theta': 1.5,'lr':0.01,'hidden_channels':64,'dropout':0.5},
'linkx': {'num_layers': 1, 'hidden_channels': 16}
},
'snap-patents': {
'mlp': {'num_layers': 2, 'hidden_channels': 256,},
'lp': [{'hops': 1, 'alpha': 0.01}, {'hops': 2, 'alpha': 0.01}],
'link': {'weight_decay': 0.001},
'sgc': [{'hops': 1, 'weight_decay': 0.001}, {'hops': 2, 'weight_decay': 0.001}],
'cs': [{'hops': 1, 'A1': 'AD', 'A2': 'AD', 'alpha1': 0.9382333382941851, 'alpha2': 0.9370796816307789}, {'hops': 2, 'A1':'DAD','A2':'DAD','alpha1':0.4501062293395896,'alpha2':4.052224260299442e-05,'scale':9.97008145139079, 'fixed': True}],
# 'gcn': {'num_layers': 2, 'hidden_channels': 32, 'lr': 0.01, 'delta':1.2},
'gcn': {'num_layers': 2, 'hidden_channels': 32, 'lr': 0.01},
'gat': {'num_layers': 2, 'hidden_channels': 12, 'gat_heads': 4, 'lr': 0.01},
'gcnjk': {'num_layers': 2, 'hidden_channels': 32, 'jk_type': 'cat', 'lr': 0.01},
'gatjk': {'num_layers': 2, 'hidden_channels': 8, 'gat_heads': 4, 'jk_type': 'max', 'lr': 0.1},
'appnp': {'hidden_channels': 256, 'gpr_alpha': 0.9, 'lr': 0.002},
'h2gcn': {'num_layers': 2, 'hidden_channels': 64, 'dropout': 0.0},
'mixhop': {'num_layers': 3, 'hidden_channels': 32, 'hops': 2},
'gprgnn': {'hidden_channels': 256, 'gpr_alpha': 0.5, 'lr': 0.01},
'gcn2': {'num_layers': 4, 'gcn2_alpha': 0.1, 'theta': 1.5,'lr':0.01,'hidden_channels':64,'dropout':0.5},
'linkx': {'num_layers': 1, 'hidden_channels': 128}
},
'wiki': {
'mlp': {'num_layers': 3, 'hidden_channels': 256,},
'lp': [{'hops': 1, 'alpha': 0.1}, {'hops': 2, 'alpha': 0.01}],
'link': {'weight_decay': 0.01},
'sgc': [{'hops': 1, 'weight_decay': 0.001}, {'hops': 2, 'weight_decay': 0.001}],
'cs': [{'hops': 1, 'A1': 'AD', 'A2': 'DAD', 'alpha1': 0.22987766570627421, 'alpha2': 0.747860051970536, 'scale': 5.187803135914824, 'fixed': True}, {'hops': 2, 'A1': 'DAD', 'A2': 'DAD', 'alpha1': 0.4379821353564854, 'alpha2': 0.9744383960028438, 'scale': 2.0087560765073786, 'fixed': True}],
# 'gcn': {'num_layers': 2, 'batch_size': 10000,'hidden_channels':128,'train_batch': 'graphsaint-rw','saint_num_steps':5,'test_num_parts':10, 'delta':1.2},
'gcn': {'num_layers': 2, 'batch_size': 10000,'hidden_channels':128,'train_batch': 'graphsaint-rw','saint_num_steps':5,'test_num_parts':10, },
'gat': {'num_layers': 2, 'hidden_channels': 32, 'gat_heads': 8, 'lr': 0.01, 'batch_size': 10000,'train_batch': 'graphsaint-rw','saint_num_steps':5,'test_num_parts':10},
'gcnjk': {'num_layers': 2, 'hidden_channels': 128, 'jk_type': 'cat', 'batch_size': 10000,'train_batch': 'graphsaint-rw','saint_num_steps':5,'test_num_parts':10},
'gatjk': {'num_layers':2,'hidden_channels': 32, 'gat_heads': 8, 'jk_type': 'max', 'batch_size': 10000,'train_batch': 'graphsaint-rw','saint_num_steps':5,'test_num_parts':10},
'appnp': {'num_layers':2,'hidden_channels': 128, 'gpr_alpha': 0.9, 'lr': 0.01, 'batch_size':10000,'train_batch': 'graphsaint-rw','saint_num_steps':5,'test_num_parts':10},
'h2gcn': {'num_layers': 2, 'hidden_channels': 64, 'dropout': 0.0,'batch_size':10000,'train_batch': 'graphsaint-rw','saint_num_steps':5,'test_num_parts':10},
'mixhop': {'num_layers': 4, 'hidden_channels': 128, 'batch_size':10000, 'hops': 2,'train_batch': 'graphsaint-rw','saint_num_steps':5,'test_num_parts':10},
'gprgnn': {'num_layers':4,'hidden_channels': 128, 'gpr_alpha': 0.1, 'batch_size':10000,'train_batch': 'graphsaint-rw','saint_num_steps':5,'test_num_parts':10},
'gcn2': {'num_layers': 8, 'lr':0.01,'hidden_channels':64,'gcn2_alpha': 0.5, 'theta': 0.5, 'batch_size':10000,'train_batch': 'graphsaint-rw','saint_num_steps':5,'test_num_parts':10},
'linkx': {'num_layers': 1, 'hidden_channels': 32,'train_batch':'row', 'num_parts':10}
},
'cora': {
'mlp': {'num_layers': 3, 'hidden_channels': 64},
'lp': [{'hops': 1, 'alpha': 0.9}, {'hops': 2, 'alpha': 0.9}],
'link': {'weight_decay': 0.01},
'sgc': [{'hops': 1, 'weight_decay': 0.001}, {'hops': 2, 'weight_decay': 0.001}],
'cs': [{'hops': 1, 'A1': 'AD', 'A2': 'DAD', 'alpha1': 0.8161683923116295, 'alpha2': 0.9456403468832562, 'scale': 7.189072082034102, 'fixed': True}, {'hops': 2, 'A1': 'DA', 'A2': 'DA', 'alpha1': 0.9674980990318431, 'alpha2': 0.8667972708007977, 'scale': 8.384236190172803, 'fixed': True}],
# 'gcn': {'num_layers':2,'weight_decay':0.01,'dropout':0.0,'hidden_channels': 64, 'lr': 0.01, 'delta':1.02},
'gcn': {'num_layers':2,'weight_decay':0.01,'dropout':0.0,'hidden_channels': 64, 'lr': 0.01, },
'gat': {'num_layers':2,'hidden_channels': 4, 'gat_heads': 4, 'lr': 0.001},
'gcnjk': {'num_layers':2,'hidden_channels': 32, 'jk_type': 'max', 'lr': 0.01},
'gatjk': {'num_layers':2,'hidden_channels': 12, 'gat_heads': 8, 'jk_type': 'cat', 'lr': 0.1},
'appnp': {'hidden_channels': 16, 'gpr_alpha': 0.1, 'lr': 0.002},
'h2gcn': {'num_layers': 2, 'hidden_channels': 8, 'dropout': 0.5},
'mixhop': {'num_layers': 2, 'hidden_channels': 32, 'hops':2},
'gprgnn': {'hidden_channels': 16, 'gpr_alpha': 0.1, 'lr': 0.002},
'gcn2': {'num_layers': 16, 'gcn2_alpha': 0.5, 'theta': 1.0,'hidden_channels':64,'dropout':0.5},
'linkx': {'num_layers': 1, 'hidden_channels': 64}
},
'amazon-photo': {
'mlp': {'num_layers': 3, 'hidden_channels': 256},
'lp': [{'hops': 1, 'alpha': 0.1}, {'hops': 2, 'alpha': 0.01}],
'link': {'weight_decay': 0.001},
'sgc': [{'hops': 1, 'weight_decay': 0.001}, {'hops': 2, 'weight_decay': 0.001}],
'cs': [{'hops': 1, 'A1': 'AD', 'A2': 'DA', 'alpha1': 0.9988066304729819, 'alpha2': 0.772003591928171, 'scale': 9.887241349003665, 'fixed': True}, {'hops': 2, 'A1': 'DA', 'A2': 'DA', 'alpha1': 0.34214593220685785, 'alpha2': 0.713075635429832, 'scale': 8.920851952543185, 'scale': 8.384236190172803, 'fixed': True}],
# 'gcn': {'num_layers':2,'weight_decay':0.01,'dropout':0.0,'hidden_channels': 64, 'lr': 0.01,'delta':1.03},
'gcn': {'num_layers':2,'weight_decay':0.01,'dropout':0.0,'hidden_channels': 64, 'lr': 0.01,},
'gat': {'num_layers':2,'hidden_channels': 12, 'gat_heads': 8, 'lr': 0.1},
'gcnjk': {'num_layers':2,'hidden_channels': 64, 'jk_type': 'max', 'lr': 0.001},
'gatjk': {'num_layers':2,'hidden_channels': 32, 'gat_heads': 8, 'jk_type': 'max', 'lr': 0.01},
'appnp': {'hidden_channels': 64, 'gpr_alpha': 0.1, 'lr': 0.002},
'h2gcn': {'num_layers': 2, 'hidden_channels': 16, 'dropout': 0.5},
'mixhop': {'num_layers': 3, 'hidden_channels': 8, 'hops':2},
'gprgnn': {'hidden_channels': 16, 'gpr_alpha': 0.2, 'lr': 0.002},
'gcn2': {'num_layers': 2, 'gcn2_alpha': 0.1, 'theta': 1.0,'hidden_channels':64,'dropout':0.5},
'linkx': {'num_layers': 1, 'hidden_channels': 16}
},
'coauthor-cs': {
'mlp': {'num_layers': 3, 'hidden_channels': 128},
'lp': [{'hops': 1, 'alpha': 0.75}, {'hops': 2, 'alpha': 0.25}],
'link': {'weight_decay': 0.001},
'sgc': [{'hops': 1, 'weight_decay': 0.001}, {'hops': 2, 'weight_decay': 0.001}],
'cs': [{'hops': 1, 'A1': 'DA', 'A2': 'DAD', 'alpha1': 0.5941172639092543, 'alpha2': 0.7848808417343226}, {'hops': 2, 'A1': 'DAD', 'A2': 'DAD', 'alpha1': 0.16260980252925394, 'alpha2': 0.8347395355309094}],
# 'gcn': {'num_layers':2,'weight_decay':0.01,'dropout':0.0,'hidden_channels': 64, 'lr': 0.01,'delta':1.0},
'gcn': {'num_layers':2,'weight_decay':0.01,'dropout':0.0,'hidden_channels': 64, 'lr': 0.01,},
'gat': {'num_layers':2,'hidden_channels': 8, 'gat_heads': 8, 'lr': 0.1},
'gcnjk': {'num_layers':2,'hidden_channels': 32, 'jk_type': 'cat', 'lr': 0.1},
'gatjk': {'num_layers':2,'hidden_channels': 12, 'gat_heads': 4, 'jk_type': 'cat', 'lr': 0.1},
'appnp': {'hidden_channels': 64, 'gpr_alpha': 0.5, 'lr': 0.002},
'h2gcn': {'num_layers': 1, 'hidden_channels': 64, 'dropout': 0.5},
'mixhop': {'num_layers': 2, 'hidden_channels': 32, 'hops':2},
'gprgnn': {'hidden_channels': 64, 'gpr_alpha': 0.5, 'lr': 0.002},
'gcn2': {'num_layers': 32, 'gcn2_alpha': 0.5, 'theta': 1.5,'hidden_channels':64,'dropout':0.5},
'linkx': {'num_layers': 2, 'hidden_channels': 128}
},
'chameleon': {
'mlp': {'num_layers': 3, 'hidden_channels': 32},
'lp': [{'hops': 1, 'alpha': 0.75}, {'hops': 2, 'alpha': 0.01}],
'link': {'weight_decay': 0.001},
'sgc': [{'hops': 1, 'weight_decay': 0.001}, {'hops': 2, 'weight_decay': 0.001}],
'cs': [{'hops': 1, 'A1': 'DA', 'A2': 'DA', 'alpha1': 0.6532450068824371, 'alpha2': 0.30175051845391043}, {'hops': 2, 'A1': 'AD', 'A2': 'DA', 'alpha1': 0.9528648880462575, 'alpha2': 0.29087232307389627}],
# 'gcn': {'num_layers': 2, 'hidden_channels': 64, 'lr': 0.005, 'dropout': 0.9, 'weight_decay': 5e-05, 'delta':1.2},
'gcn': {'num_layers': 2, 'hidden_channels': 64, 'lr': 0.005, 'dropout': 0.9, 'weight_decay': 5e-05, },
'gat': {'num_layers': 2, 'hidden_channels': 32, 'gat_heads': 8, 'lr': 0.01, 'weight_decay':1e-05, 'dropout':0.5},
'gcnjk': {'num_layers': 2, 'hidden_channels': 64, 'jk_type': 'max', 'lr': 0.01, 'weight_decay':1e-05, 'dropout':0.0},
'gatjk': {'num_layers': 2, 'hidden_channels': 32, 'gat_heads': 8, 'jk_type': 'max', 'lr': 0.01, 'weight_decay':5e-05, 'dropout':0.5},
'appnp': {'hidden_channels': 16, 'gpr_alpha': 0.1, 'lr': 0.005, 'weight_decay':1e-05, 'dropout':0.9},
'h2gcn': {'num_layers':2, 'hidden_channels':64, 'dropout':0.5, 'lr':0.005,'weight_decay':1e-05},
'mixhop': {'num_layers': 2, 'hidden_channels': 32, 'lr': 0.01, 'weight_decay': 1e-05, 'dropout':0.9,'hops': 2},
'gprgnn': {'hidden_channels': 16, 'gpr_alpha': 0.9, 'lr': 0.01, 'weight_decay': 5e-05, 'dropout':0.9},
'gcn2': {'num_layers': 2, 'hidden_channels':64,'gcn2_alpha': 0.1, 'theta': 1.5, 'lr': 0.01, 'weight_decay': 5e-05, 'dropout':0.5},
'linkx': {'num_layers': 3, 'hidden_channels': 32, 'lr': 0.005, 'weight_decay': 0.01, 'dropout':0.5}
},
'squirrel': {
'mlp': {'num_layers': 3, 'hidden_channels': 256},
'lp': [{'hops': 1, 'alpha': 0.75}, {'hops': 2, 'alpha': 0.01}],
'link': {'weight_decay': 0.001},
'sgc': [{'hops': 1, 'weight_decay': 0.001}, {'hops': 2, 'weight_decay': 0.001}],
'cs': [{'hops': 1,'A1': 'AD', 'A2': 'DA', 'alpha1': 0.3571971400165756, 'alpha2': 0.8574731781604191, 'scale': 6.710984978278087}, {'hops': 2, 'A1': 'DAD', 'A2': 'DA', 'alpha1': 0.5694114031396995, 'alpha2': 0.8749544103470049}],
# 'gcn': {'num_layers': 2, 'hidden_channels': 64, 'lr': 0.01, 'dropout': 0.8, 'weight_decay': 0.0001},
'gcn': {'num_layers': 2, 'hidden_channels': 64, 'lr': 0.01, 'dropout': 0.8, 'weight_decay': 0.0001, 'col': True, 'adj1': True},
'gat': {'num_layers': 2, 'hidden_channels': 32, 'gat_heads': 8, 'lr': 0.01, 'weight_decay':0.0001, 'dropout':0.9, 'col': True, 'adj1': True},
'gcnjk': {'num_layers': 2, 'hidden_channels': 64, 'jk_type': 'max', 'lr': 0.01, 'weight_decay':5e-05, 'dropout':0.5, 'col': True, 'adj1': True},
'gatjk': {'num_layers': 2, 'hidden_channels': 32, 'gat_heads': 8, 'jk_type': 'max', 'lr': 0.005, 'weight_decay':1e-05, 'dropout':0.5, 'col': True, 'adj1': True},
'appnp': {'hidden_channels': 16, 'gpr_alpha': 0.1, 'lr': 0.005, 'weight_decay':1e-05, 'dropout':0.9, 'col': True, 'adj1': True},
'h2gcn': {'num_layers': 2, 'hidden_channels': 64, 'dropout': 0.5, 'lr': 0.01,'weight_decay':1e-05, 'col': True, 'adj1': True},
'mixhop': {'num_layers': 2, 'hidden_channels': 16, 'lr': 0.01, 'weight_decay': 5e-05, 'dropout':0.9,'hops': 2, 'col': True, 'adj1': True},
'gprgnn': {'hidden_channels': 16, 'gpr_alpha': 0.1, 'lr': 0.01, 'weight_decay': 0.001, 'dropout':0.9, 'col': True, 'adj1': True},
'gcn2': {'num_layers': 2, 'hidden_channels':64,'gcn2_alpha': 0.5, 'theta': 1.5, 'lr': 0.005, 'weight_decay': 5e-05, 'dropout':0.5, 'col': True, 'adj1': True},
'linkx': {'num_layers': 1, 'hidden_channels': 64, 'lr': 0.005, 'weight_decay': 0.005, 'dropout':0.5}
},
'ogbn-products': {
'gcn': {'num_layers': 2, 'batch_size': 10000,'hidden_channels':128,'train_batch': 'graphsaint-rw','saint_num_steps':5,'test_num_parts':10, }
}
}