ki-ljl / GNNs-for-Link-Prediction

Some GNNs are implemented using PyG for link prediction tasks, including: GCN, GraphSAGE, GAT, Node2Vec、RGCN、HGT and HAN, which will continue to be updated in the future.

GNNs-for-Link-Prediction

Some GNNs are implemented using PyG for link prediction tasks, including: GCN, GraphSAGE, GAT, Node2Vec、RGCN、HGT and HAN, which will continue to be updated in the future.

Dependencies

numpy==1.22.4
pandas==1.3.3
scikit-learn==1.2.1
torch==1.12.1+cu113
torch-cluster==1.6.0+pt112cu113
torch-geometric==2.2.0
torch-scatter==2.1.0+pt112cu113
torch-sparse==0.6.16+pt112cu113
torch-spline-conv==1.2.1+pt112cu113
tqdm==4.62.3