支持异构图、集成GraphGym,超好用的图神经网络库PyG更新2.0版本( 二 )


支持异构图、集成GraphGym,超好用的图神经网络库PyG更新2.0版本
文章图片
重大改变
datasets.AMiner数据集现在返回data.HeteroData对象;transforms.AddTrainValTestMask已被替换为transforms.RandomNodeSplit;由于data.Data的存储布局发生了显着变化以支持异构图 , 因此需要通过删除root/processed文件夹来重新处理已处理的数据集 。
此外 , data.Data.__cat_dim__、data.Data.__inc__现在需要额外的参数输入:
def__cat_dim__(self,key,value,*args,**kwargs):
pass
def__inc__(self,key,value,*args,**kwargs):
pass
如果你修改了自定义data.Data对象中的__cat_dim__或__inc__ , 请确保应用上述更改 。
了解更多变化 , 请参考原项目 。
核心团队成员
PyG核心团队共有6位成员 , 包括多特蒙德工业大学四年级图机器学习博士生MatthiasFey、多特蒙德工业大学博士生JanEricLenssen(核心贡献者)和图网络领域大牛&斯坦福计算机科学副教授JureLeskovec(担任顾问) 。
支持异构图、集成GraphGym,超好用的图神经网络库PyG更新2.0版本
文章图片
从左至右依次为MatthiasFey、JanEricLenssen和JureLeskovec 。
此外还有三位华人成员 , 分别是斯坦福大学计算机科学四年级博士生尤佳轩(JiaxuanYou , 核心领导者)、斯坦福大学计算机科学博士生RexYing(核心贡献者 , 他将于2022年加入耶鲁大学担任助理教授)和卡内基梅隆大学-汉斯学院三年级博士生赵越(YueZhao , 核心贡献者) 。
支持异构图、集成GraphGym,超好用的图神经网络库PyG更新2.0版本
文章图片
从左至右依次为尤佳轩、RexYing和赵越 。
日前 , 赵越在知乎分享了自己参与PyG2.0版本设计与升级过程中的心得体会 , 感兴趣的读者可以参考阅读 。
知乎链接:https://www.zhihu.com/pin/1420674585365196800
参考链接:
http://www.pyg.org/
来源:机器之心Pro