yuyc1729 / RGCN-tensorflow2-dgl

We created RGCN and MPNN models using tensorflow2 and dgl for link prediction.
1 stars 0 forks source link

RGCN-tensorflow2-dgl

We created RGCN and MPNN models using tensorflow2 and dgl for link prediction.
FB237k-15 and wn18 datasets can be loaded using labelled_loader.py in a form which is consistent with the one in dgl.