OpenHGNN发布0.7版本

文章预览

OpenHGNN现已上线最新的0.7版本，欢迎大家从启智社区、GitHub或通过pip下载使用！OpenHGNN是一个基于 DGL和 PyTorch的开源异质图神经网络工具包，集成了异质图神经网络的前沿模型。新版本我们增加了10个新的异质图神经网络模型，包含基于异质图提示的预训练pipeline。OpenHGNN还在新版本添加了支持dgl.sparse的模型API，支持更便捷的图神经网络聚合操作。此外，新版本还新增了对DGL.graphbolt的数据集成和模型训练流程，支持更通用的数据处理框架。最后，我们新增了分布式训练流程，采用数据并行化方式，在多个显卡上同时进行训练。更加充分地使用了算力资源，提高了计算速度。 一、新增图神经网络模型、数据集 OpenHGNN 0.7 版本新增10种异质图模型算法 新增保留网络模式的多任务学习异质网络模型NSHE [IJCAI 2020] 新增异质图预训练提示学习模型HGPrompt [AAAI 2024 ………………………………