pyg-team/pytorch_geometric

HANConv link prediction example

Open

#4,221 opened on 2022幎3月8日

GitHub で芋る
 (4 comments) (0 reactions) (0 assignees)Python (3,514 forks)batch import
1 - Priority P1documentationfeaturehelp wanted

Repository metrics

Stars
 (19,985 stars)
PR merge metrics
 (平均マヌゞ 35d 1h) (30d で 14 merged PRs)

説明

📚 Describe the documentation issue

There is an example for how to use HANConv to do node classification: https://github.com/pyg-team/pytorch_geometric/blob/master/examples/hetero/han_imdb.py, and we also have an example for how to do link prediction with heterogeneous graph: https://github.com/pyg-team/pytorch_geometric/blob/master/examples/hetero/hetero_link_pred.py.

However, we have to do different transformations in the two examples. It is tricky to figure out how to do link prediction with HANConv.

Suggest a potential alternative/fix

I would suggest to have an example showing how to do link prediction with HANConv, especially how to do T.RandomLinkSplit and T.AddMetaPaths together.

コントリビュヌタヌガむド