Described in #167, ConvKB loads trained model params from a dict with keys: ['ent_emb', 'rel_emb'] instead of a list (indexed: [0, 1]).
There's no reason not to use a dictionary to save trained model parameters - the overhead is negligible and code is easier to read rather than interpreting what different indices of a list are supposed to map to for each model (I realize for most of the models it's just 0 -> entities, 1->relations, but this will get more complicated).
Description
Described in #167, ConvKB loads trained model params from a dict with keys: ['ent_emb', 'rel_emb'] instead of a list (indexed: [0, 1]).
There's no reason not to use a dictionary to save trained model parameters - the overhead is negligible and code is easier to read rather than interpreting what different indices of a list are supposed to map to for each model (I realize for most of the models it's just 0 -> entities, 1->relations, but this will get more complicated).
Fix
Change
EmbeddingModel
:get_embeddings()
: