search

alibaba / graph-learn

An Industrial Graph Neural Network Framework
Apache License 2.0
1.28k stars 267 forks source link

如何使用pytorch编写GNN模型，该如何处理EgoGraph？ #147

Closed Ningsir closed 3 years ago

jackonan commented 3 years ago

EgoGraph是模型相关的，目前模型层接口只内置了TF的实现，没有PyTorch。如果用PyTorch，可以用GSL返回的数据，自己实现后续部分。GraphLearn 2.0接口也正在发布的路上，届时接口会更清晰易懂。

baoleai commented 3 years ago

https://graph-learn.readthedocs.io/en/latest/algo/torch/index.html