torch-geometric - 使用 PyTorch Geometric 构建图神经网络
指导用户使用 PyG 表示图数据、构建图神经网络,并处理节点分类、图分类、链接预测和异构图任务。
标签
更新于: 2026-10-02能力
典型输入
典型输出
该技能可以做什么
- 表示图数据
- 构建图神经网络
- 实现消息传递
- 训练节点分类器
- 执行图分类
- 执行链接预测
- 处理异构图
- 应用图变换
输入
- 图神经网络任务需求
- 图数据
- 节点特征
- 边特征
- 标签和掩码
输出
- 实现指导
- PyTorch Geometric 代码
- 模型配置
- 训练方案
- 评估方案
要求
- PyTorch 环境
- torch_geometric 软件包
