名称 | 定义 | 参数值 |
设配 | Device | Gpu |
图卷积层数 | GCN_Layers | 2 |
输入维度 | Input_dimension | 2000 |
隐藏层维度 | Hidden_Layer_dimension | 16 |
输出层类别 | Output_Layer_category | 4 |
权重衰减 | Weight_decay | 5e−4 |
训练迭代次数 | Epochs | 1000 |
学习率 | Learning_rate | 0.04 |
损失函数 | Loss | 交叉熵 |
优化器 | optimizer | Adam |