Skip to content

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.

Notifications You must be signed in to change notification settings

ki-ljl/GNNs-for-Link-Prediction

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

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.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages