Question about running transe

mana-ysh / knowledge-graph-embeddings

Implementations of Embedding-based methods for Knowledge Base Completion tasks

Apache License 2.0

257 stars 63 forks source link

Question about running transe #5

Closed shaoyx closed 6 years ago

shaoyx commented 6 years ago

Hi all, I ran TransE on WordNet with following command: python train.py --mode pairwise --ent ../dat/wordnet-mlj12/train.entlist --rel ../dat/wordnet-mlj12/train.rellist --train ../dat/wordnet-mlj12/wordnet-mlj12-train.txt --valid ../dat/wordnet-mlj12/wordnet-mlj12-valid.txt --method transe But the loss is not converging.

Do I have set the wrong parameters?

mana-ysh commented 6 years ago

Hi @simon0227

Thank you for reporting. It seems that it's my mistake. I'll check and fix my code.

mana-ysh commented 6 years ago

This is due to the incorrect score calculation and I fixed it. Please retry to run TransE. it should work and converge the training loss.

shaoyx commented 6 years ago

It works now. Thanks~