PyG-GCN_res-CS
这是使用C&S方法对模型的改进。
ogbn-arxiv
出模型:
查看C&S方法:
改善策略:
添加C&S方法
环境要求
pytorch == 1.7.1
pytorch_geometric == 1.6.3
ogb == 1.2.4
实验设置:
该模型为8层,共运行10次,共得出500个纪元。
python gcn_res_cs.py
详细的超参数:
num_layers = 8
hidden_dim = 128
dropout = 0.5
lr = 0.01
runs = 10
epochs = 500
alpha = 0.2
beta = 0.7
num_correction_layers = 50
correction_alpha = 0.8
num_smoothing_layers = 50
smoothing_alpha =
2023-04-10 22:34:19
7KB
Python
1