Pytorch_geometric: link prediction

[Purpleheather]

❓ Questions & Help

Is there an example of RGCN doing link prediction?

[rusty1s]

No, we sadly do not have a link prediction example of RGCN, but it should be relatively easy to implement. The main part would be to add the necessary datasets to the repository. Feel free to work on it if you like :)

[WMF1997]

hello @Purpleheather

Here is a repo about link prediction in github.
NOTE THAT it is implemented on TensorFlow 1.7.0 and networkx 2.0 (the repo's owner shows that networkx 2.0+ does not work, in this issue)

and its arxiv preprint paper: Link Prediction Based on Graph Neural Networks

I installed it and run it on the environment below:
Ubuntu 16.04 (NO CUDA)
Python 3.7
TensorFlow 1.13.1

I do not know your environment. And I run the repo's gnn_implementation folder, whose main.py uses mutag dataset. (in fact, we have mutag dataset in pyg)

in ./GNN_implement/gnn.py: 118-131 (118-131 lines) shows its gnn network. I think it is a GCN.

hope to help you a little.

yours sincerely
@wmf1997

[tchaton]

@WMF1997, interesting to follow this thread. I might give a try to SEAL.

Best,
T.C

[tchaton]

Here is the pytorch version: https://github.com/muhanzhang/SEAL

[laifi]

:eyes: :eyes: :arrow_right: https://github.com/rusty1s/pytorch_geometric/blob/master/examples/link_pred.py :tada:

[DavideBuffelli]

👀 👀 ➡️ https://github.com/rusty1s/pytorch_geometric/blob/master/examples/link_pred.py 🎉

Hi, I think there are several errors in this Link Prediction example. I opened an issue about it here https://github.com/rusty1s/pytorch_geometric/issues/1121

[dongkwan-kim]

👀 👀 ➡️ /examples/link_pred.py@master 🎉

Hi, I think there are several errors in this Link Prediction example. I opened an issue about it here #1121

It is fixed now 😊 #1124