基于GCN的Cora引文网络数据分类

一、环境安装

• python 版本：3.10
• 安装命令：pip install -r requirements.txt

  二、数据准备

• 数据放在项目 /release/data 目录下

  三、训练

• 先进入 code 目录再输入 python train.py
• 训练时：

    1.首先开启GPU（默认开启，可以关闭，设置为1是开启,0是关闭），设置全局随机数种子（默认为42，可更改）；
    2.加载数据，load_data()负责加载数据，需要根据自己数据实践存放的位置，把路径传给参数data_path；
    3.进行特征归一化、图的边归一化以及稀疏格式转换；
    4.初始化模型（GCN）和优化器（Adam）；
    5.正式开始训练（训练轮次epoch可以更改），根据验证集的评分（指标为准确率）保存最佳模型

  四、评测/推理

• 先进入 code 目录，再输入 python test.py
• 测试时，自动调用最佳模型进行测试
• 测试结果会保存为 result.json 文件（放在 Results 目录下）

  五、结果说明、

• 指标名称与计算方式：准确率（acc）；预测对的结点数/总预测结点数