This is the source code of the Rotate4D algorithm of the "Knowledge Graph Embedding with the Special Orthogonal Group in Quaternion Space for Link Prediction" paper.
Thanh Le, Huy Tran, Bac Le (Faculty of Information Technology, University of Science, Ho Chi Minh City, Vietnam and Vietnam National University, Ho Chi Minh City, Vietnam)
- pytorch >= 1.0.0 (except version 1.5.0)
- numpy >= 1.15.4 #For data preparation process
- scikit-learn >= 0.20.2