GammaGL发布0.5版本

文章预览

GAMMA LAB于今日正式发布GammaGL的0.5版本，大家可以从GitHub，启智社区或通过pip安装最新的gammagl。在新版本中引入了10种图神经网络模型，5种新数据集以及优化部分问题。特别地，我们增加了GraphStore和FeatureStore，使用户能够从nebula和neo4j数据库中加载数据；为了提升采样效率，我们实现了CUDA上的采样方法，较原始方法效率提升了9 ~ 22 倍；同时为了应对更广泛的数据场景，我们扩展了segment算子以支持更多的数据类型，我们也实现了bspmm算子，效率较之前的方法提升2.3倍左右。 一、新增图神经网络模型、数据集和工具类 GammaGL第五个版本支持70余种模型算法供大家使用，包括新增10种同质图模型算法和异质图模型算法，以及优化部分已有模型。 新增带有保真度项的通用扩散方程框架Hid-Net [AAAI 2023] 新增从协调计算、IO和内存角度优化计算图的图神经网络模 ………………………………