search

malllabiisc / CompGCN

ICLR 2020: Composition-Based Multi-Relational Graph Convolutional Networks
Apache License 2.0
597 stars 107 forks source link

How to get trained embedding matrix #27

Open SRL94 opened 3 years ago

SRL94 commented 3 years ago

Hi, nice work. How do I get the trained embeddings of entities and relations?

svjan5 commented 3 years ago

As you can see from Figure 2 in the paper. The output of CompGCN will give you the trained entity and relation embeddings image

SRL94 commented 3 years ago

Thank you, but the foward gives the scores. Do you mean the outputs of forward_base are entity and relation embeddings?

ralgond commented 1 year ago

Hi, do you have resolved this question?