awslabs / dgl-ke

High performance, easy-to-use, and scalable package for learning large-scale knowledge graph embeddings.
https://dglke.dgl.ai/doc/
Apache License 2.0
1.28k stars 196 forks source link

Add decoders to new framework #208

Closed zby123 closed 3 years ago

zby123 commented 3 years ago

Issue #, if available:

Description of changes: Add TransR, DistMult, ComplEx, RESCAL, RotatE, SimplE decoders to new framework.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.