torch-geometric - 使用 PyG 构建和训练图神经网络
用于节点与图分类、链接预测、分子属性预测及几何数据的图神经网络开发与训练。
标签
更新于: 2026-10-08能力
典型输入
典型输出
该技能可以做什么
- 构建图神经网络
- 执行节点与图分类
- 预测图链接
- 处理异构图
- 批处理不规则图数据
- 实现消息传递层
- 加载基准数据集
- 创建自定义数据集
- 训练大规模图模型
输入
- 图数据
- 节点特征
- 边索引
- 标签
- 数据集文件
- 学习任务规格
输出
- 模型预测结果
- 训练后的模型参数
- 处理后的图数据集
- 批处理图对象
要求
- Python 环境
- PyTorch 安装
- torch-geometric 安装
