/ProjE

Embedding Projection for Knowledge Graph Completion

Primary LanguagePython

ProjE

Embedding Projection for Knowledge Graph Completion


How to run the code

First, please download the data file from https://github.com/thunlp/KB2E and unzip the data.zip file. You will get two folders FB15k and WN18.

You can call ProjE using

./ProjE_softmax.py --dim 200 --batch 200 --data ./data/FB15k/ --eval_per 1 --worker 3 --eval_batch 500 --max_iter 100 --generator 10