Implementation of TransE [1] model in PyTorch.
Source/Metric | Hits@1 (raw) | Hits@3 (raw) | Hits@10 (raw) | MRR (raw) |
---|---|---|---|---|
Paper [1] | X | X | 34.9 | X |
TransE-PyTorch | 11.1 | 25.33 | 46.53 | 22.29 |
python3 train.py --dataset_path=<path_to_your_dataset> --model_dir=<path_to_your_model_params>