tf_geometric是一个高效且友好的图神经网络库,同时支持TensorFlow 1.x 和 2.x。tf_geometric今日发布了0.0.58版,支持稀疏节点特征。新特性大幅度简化了featureless场景的GNN实现。
项目简介
tf_geometric是一个高效且友好的图神经网络库,同时支持TensorFlow 1.x 和 2.x。
受到rusty1s/pytorch_geometric项目的启发,我们为TensorFlow构建了一个图神经网络(GNN)库。tf_geometric 同时提供面向对象接口(OOP API)和函数式接口(Functional API),你可以用它们来构建有趣的模型。
0.0.58更新内容
支持稀疏节点特征。新特性大幅度简化了featureless场景的GNN实现。下面的示例展示了用稀疏节点特征(特征是节点ID的One-hot编码)进行图卷积操作: