pyg-team/pytorch_geometric

HANConv link prediction example

Open

#4221 aperta il 8 mar 2022

Vedi su GitHub
 (4 commenti) (0 reazioni) (0 assegnatari)Python (3514 fork)batch import
1 - Priority P1documentationfeaturehelp wanted

Metriche repository

Star
 (19.985 star)
Metriche merge PR
 (Merge medio 35g 1h) (14 PR mergiate in 30 g)

Descrizione

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

Guida contributor