Skip to content

vardaan123/TransE_wikidata

Repository files navigation

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

About

Code to train TransE embeddings on wikidata

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages