Link Prediction Models
Source code
This repository contains the following link prediction models:
- transE
- DistMult
- convE
- simplE
- rotatE
Dataset:
-
FB13
-
FB15K
-
FB15K237
-
WN11
-
WN18
-
WN18RR
Prerequisites
- Python3
- Pytorch 1.0+
- faiss(for evaluation)