Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Settings in FB15k #7

Closed
huxiao-neu opened this issue Feb 9, 2018 · 1 comment
Closed

Settings in FB15k #7

huxiao-neu opened this issue Feb 9, 2018 · 1 comment

Comments

@huxiao-neu
Copy link

hey!
To get the best results on the FB15k, what should the parameters be set to? I use the same settings as the FB15k-237 and can only get hit@10: 86 %.

@TimDettmers
Copy link
Owner

Make sure that you run it for a long time. You will see gradual improvement over time. I ran the model that I reported in the paper for 2 days I think. At first, it may seem to stagnate but still improves slowly.

I used the following parameters:

  • input dropout 0.2
  • feature dropout 0.2
  • hidden dropout 0.3
  • learning rate 0.003
  • learning rate decay 0.99
  • label smoothing 0.1 (I think, this is not clear from my logs)

I will close this for now. Let me know if you have any more problems with replication.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants