Skip to content

Double Permutation Equivariance for Knowledge Graph Completion

License

Notifications You must be signed in to change notification settings

PurdueMINDS/ISDEA

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

(UPDATED) Double Equivariance for Inductive Link Prediction for Both New Nodes and New Relation Types

This repository contains the code for three (3) models introduced in the paper "Double Equivariance for Inductive Link Prediction for Both New Nodes and New Relation Types". These models are located in their respective directory as follows:

  • ISDEA/: The legacy implementation of the Inductive Structural Double Equivariant Architecture (ISDEA).
  • ISDEA_plus/: The new and improved Inductive Structural Double Equivariant Architecture Plus (ISDEA+), which achieves 20x - 120x speedup and attains superior performance compared to the legacy ISDEA.
  • DEq_InGram: The Double Equivariant version of InGram, which achieves better performance than the original InGram on the doubly inductive link prediction task on knowledge graphs considered in the paper.

Please refer to the README.md file in each directory for more details on how to run the code.

About

Double Permutation Equivariance for Knowledge Graph Completion

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 3

  •  
  •  
  •