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

embeddings initialization #177

Closed Antares-97 closed 3 years ago

Antares-97 commented 3 years ago

Hi! Is there a way to initialize the embeddings from external files? And how can I initialize the node and relation embeddings?

classicsong commented 3 years ago

Right now you have to change the code to add support for loading your own node and relation embeddings. You can do that after here: https://github.com/awslabs/dgl-ke/blob/b34bf1d538a3a2184a4e767aeaa3998ed290233e/python/dglke/train_pytorch.py#L281