1.环境安装

• Python版本

  python==3.10.20

• 其他关键依赖

  jittor==1.3.10.0
  jittor_geometric==2.0.0
  numpy==1.24.4

• 安装命令

  pip install python==3.10.20 jittor==1.3.10.0 jittor_geometric==2.0.0 numpy==1.24.4

2.数据准备

• 数据集说明

数据集文件 data/cora.pkl 为 pickle 格式，包含以下字段：

字段	类型	说明
x	numpy array (2708, 1433)	节点特征矩阵
y	numpy array (2708,)	节点标签（测试集标签为 -1）
edge_index	numpy array (2, num_edges)	边列表
train_mask	numpy bool array (2708,)	训练集掩码
val_mask	numpy bool array (2708,)	验证集掩码
test_mask	numpy bool array (2708,)	测试集掩码
num_classes	int	类别数（7）
num_features	int	特征维度（1433）

3.训练

运行脚本在scripts/main.py

• 运行命令

python scrpits/main.py

4.评测/推理

终端输出最终预测准确率best_val_acc

若准确率达到比赛要求，则通过

5.结果说明

结果输出并保存在result.json文件中

结果集文件result.json字段说明：

字段	类型	说明
Key	字符串	测试集节点的编号
Value	整数，0-6	预测的类别编号