torch-geometric - 使用 PyG 开发图神经网络
使用 PyTorch Geometric 开发图神经网络,处理图学习、几何数据、分子性质预测和异构图任务。
标签
更新于: 2026-10-08该技能可以做什么
- 创建图数据
- 加载基准数据集
- 构建图神经网络模型
- 实现消息传递层
- 批处理图数据
- 训练图模型
- 处理异构图
- 处理分子图
输入
- 图数据
- 节点特征
- 边特征
- 任务标签
- 数据集文件
- 学习任务
输出
- 图预测结果
- 节点预测结果
- 链接预测结果
- 训练后的图神经网络模型
- 图嵌入
- 处理后的图数据集
要求
- Python 环境
- PyTorch
- torch_geometric 软件包
