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 offline inference support #103

Closed classicsong closed 4 years ago

classicsong commented 4 years ago

Add support for offline inference of two different tasks:

For linkage score ranking, we support TransE_l1, TransE_l1, DistMult, ComplEx, RESCAL, RotatE score functions. We provides two different score funcs:

For embedding similarity ranking, we support five different similarity funcs: