lib-torch-geometric - 使用 PyG 开发图神经网络
使用 PyTorch Geometric 开发和训练图神经网络，用于图学习、几何数据、异构图和分子属性预测。
标签
更新于: 2026-10-07该技能可以做什么
- 执行节点和图分类
- 预测图链接
- 构建 GCN 模型
- 构建 GAT 模型
- 构建 GraphSAGE 模型
- 处理异构图
- 训练大规模图模型
输入
- 图节点特征
- 图边连接关系
- 边特征
- 目标标签
- 图数据集
- 分子结构
- 任务规格
输出
- 图数据对象
- 训练后的图神经网络模型
- 节点分类结果
- 图分类结果
- 链接预测结果
- 分子属性预测结果
要求
- Python 环境
- PyTorch 安装
- PyTorch Geometric 安装