QubitE

code for "QubitE: Qubit Embedding for Knowledge Graph Completion".

Environment

• PyTorch 1.8.1 + cuda 10.2

How to Run

QubitE with $\psi != 0$ :

python train_QubitE.py --name="QubitE"

QubitE with $\psi = 0$ :

python train_QubitE2.py --name="QubitE2"