search

haseebs / OWE

Pytorch code for An Open-World Extension to Knowledge Graph Completion Models (AAAI 2019)
https://aaai.org/ojs/index.php/AAAI/article/view/4162
37 stars 9 forks source link

Pretrained ComplEx Embeddings for DBpedia #22

Open JulianSe26 opened 2 years ago

JulianSe26 commented 2 years ago

Hi there,

is there a chance to get the pretrained ComplEx Embeddings for DBPedia50k, similar as for the FB-OWE dataset ?

Thank you

haseebs commented 2 years ago

Unfortunately, I no longer have these embeddings. However, you can easily obtain them using the provided script in the open_ke folder.

Make sure that you are using the correct open_ke version: https://github.com/thunlp/OpenKE/tree/0a55399b3e800bc779582c4784cac96f00230fd8

JulianSe26 commented 2 years ago

many thanks

JulianSe26 commented 2 years ago

Are you sure the link contains the correct open_ke version ? As I used your script for training the model, I get several errors that some methods of the open_ke config class are not available, and indeed they are not implemented (like con.set_test_link_prediction(True) or con.set_log_on(1) , any clue what went wrong ?

haseebs commented 2 years ago

The version of openKE should be correct as it is what I used. However, it seems that I forgot to update the dbpedia50 openKE training scripts.

For now, you can try using the FB-15k-237 openKE training files, I believe that they should work. If so, you can simply copy the hyper-parameters and the paths from the dbpedia50 script to the FB-15k-237 ones.

I would recommend re-tuning the dbpedia50k closed-world KGC as I am not sure whether those are the best hyper parameters anymore.

I will look through my old hard-drive later to see whether I have a backup of the trained dbpedia50k embeddings. It is unlikely, but I will update here if I find them.

JulianSe26 commented 2 years ago

Thanks for your continuous support ! Really appreciate it 👍