Skip to content
/ r-gcn Public

PyToch implementation of R-GCN model for node classification and link prediction

Notifications You must be signed in to change notification settings

kkteru/r-gcn

Repository files navigation

r-gcn

PyToch implementation of R-GCN model for node classification and link prediction

  • Code skeleton
  • Get link_data to DistMul properly
  • Get the adjacency matrix to DistMul
  • Pipeline the two experiments somehow
  • Test training loop for classification task
  • Test training loop for link prediction
  • Complete the evaluation scripts to save checkpoints and best model
  • Evaluation scripts
  • Revamp GCN into configurable layers
  • Refactor into Encoder and Decoder hierarchical structure with Aggregator

Requirements: rdflib, wget, scipy, sklearn

About

PyToch implementation of R-GCN model for node classification and link prediction

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published