search

BaseModelAI / cleora

Cleora AI is a general-purpose model for efficient, scalable learning of stable and inductive entity embeddings for heterogeneous relational data.
https://cleora.ai
Other
485 stars 51 forks source link

Initialize with node features #26

Open kmichael08 opened 3 years ago

kmichael08 commented 3 years ago

Is it possible to leverage the information of the node features, e.g. initialize the embeddings?

barbara3430 commented 3 years ago

Hi Michał, nice to hear from you :) It makes sense to initialize Cleora with precomputed node features instead of random initialization. This way, the obtained node embeddings will keep the information prior induced by the initial embeddings. During training, graph-based knowledge will be additionally incorporated within the embeddings. You can attempt to load precomputed vectors as input, or wait for our implementation of easy custom embedding loading which is due in April :)

kmichael08 commented 3 years ago

Great, thanks! Looking forward to seeing this feature :)