How to use the NBFNet to predict a candidate list?

KiddoZhu / NBFNet-PyG

PyG re-implementation of Neural Bellman-Ford Networks (NeurIPS 2021)

MIT License

61 stars 13 forks source link

Open ralgond opened 1 year ago

ralgond commented 1 year ago

hi, this repo is a nice job.

But, I can not find a method to make the NBFNet to predict. For example, given a (head, relation) and get a list candidates(tails).