mana-ysh / knowledge-graph-embeddings

Implementations of Embedding-based methods for Knowledge Base Completion tasks

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Question about running transe

shaoyx opened this issue · comments

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.
image

Do I have set the wrong parameters?

Hi @simon0227

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

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.

It works now. Thanks~