search

BaseModelAI / cleora

Cleora AI is a general-purpose open-source model for efficient, scalable learning of stable and inductive entity embeddings for heterogeneous relational data. Created by Synerise.com team.
https://cleora.ai
Other
495 stars 53 forks source link

Initialize with node features #26

Closed kmichael08 closed 1 week 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 :)