Skip to content

Commit

Permalink
Update knowledge_graph_embedding.md
Browse files Browse the repository at this point in the history
add dismult png source: [dglke](https://dglke.dgl.ai/doc/kg.html)
  • Loading branch information
robertdhayanturner authored Jan 5, 2024
1 parent cf42080 commit bddca75
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions docs/use_cases/knowledge_graph_embedding.md
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,7 @@ KGE algorithms vary in the similarity functions they employ, and how they define
For our KGE model demo, we opted for the DistMult KGE algorithm. It works by representing the likelihood of relationships between entities (i.e., similarity) as a _bilinear_ function. Essentially, DisMult KGE assumes that the score of a given triple (comprised of a head entity $h$, a relationship $r$, and a tail entity $t$) can be computed as: $h^T \text{diag}(r) t$.

![DistMult similarity function](../assets/use_cases/knowledge_graph_embedding/distmult.png)
source: [dglke](https://dglke.dgl.ai/doc/kg.html)

The model parameters are learned (internalizing the intricate relationships within the KG) by _minimizing cross entropy between real and corrupted triplets_.

Expand Down

0 comments on commit bddca75

Please sign in to comment.