search

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

What is the ID of the returned node embedding? #171

Closed yeqing97 closed 3 years ago

yeqing97 commented 3 years ago

After training with the train.tsv, valid.tsv and test.tsv, I got the node embedding named XXX_YYY_entity.npy in the ckpts folder. How can I relate the node embedding with my input entity?