r-gcn icon indicating copy to clipboard operation
r-gcn copied to clipboard

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

r-gcn

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

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

Requirements: rdflib, wget, scipy, sklearn