目录

赛道一热身赛

使用说明

  1. 配置环境:参考 JittorGeometric 官方安装指南,安装 Jittor、JittorGeometric 和依赖。
  2. 数据集文件 cora.pkl 应放置于 data/ 目录下
  3. 运行 python gcn.py 进行训练和预测

数据集说明

数据集文件 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)
关于
34.0 KB
邀请码
    Gitlink(确实开源)
  • 加入我们
  • 官网邮箱:gitlink@ccf.org.cn
  • QQ群
  • QQ群
  • 公众号
  • 公众号

版权所有:中国计算机学会技术支持:开源发展技术委员会
京ICP备13000930号-9 京公网安备 11010802032778号