pyg-team/pytorch_geometric

HANConv link prediction example

Open

#4,221 创建于 2022年3月8日

在 GitHub 查看
 (4 评论) (0 反应) (0 负责人)Python (3,514 fork)batch import
1 - Priority P1documentationfeaturehelp wanted

仓库指标

Star
 (19,985 star)
PR 合并指标
 (平均合并 35天 1小时) (30 天内合并 14 个 PR)

描述

📚 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.

贡献者指南