1、准备工作
按照JK-Nets的paper中的划分方式,训练集(60%) 、验证集(20%)、测试集(20%)。
数据集划分(数量) |
训练集(60%) |
验证集(20%) |
测试集(20%) |
Cora |
1624 |
542 |
542 |
Citeseer |
1995 |
666 |
666 |
Pubmed |
11829 |
3944 |
3944 |
超参数也严格遵循了paper中的取值。
超参数等训练因素 |
取值/范围 |
loss_func |
NLLLoss |
optimizer |
Adam |
lr |
0.005 |
dropout |
0.5 |
hidden_dim |
16 |
weight_decay |
5e-4 |
epoch |
100 |