search

uma-pi1 / kge

LibKGE - A knowledge graph embedding library for reproducible research
MIT License
765 stars 124 forks source link

generate embeddings #277

Closed halimaaol closed 11 months ago

halimaaol commented 1 year ago

Hi I have generated a chekpoint.pt for a dataset. and I want to generate embeddings for a specific list of entities. can you help me to generate embeddings ?

rgemulla commented 1 year ago

Have a look at here on how to work with a checkpoint. Given model in that code, use the models embedders; e.g. model.get_s_embedder().embed(indexes), where indexes is an array holding the indexes of the entities for which you'd like to extract embeddings.