If you use the code, please kindly cite the following paper:
Yankai Lin, Zhiyuan Liu, Maosong Sun, Yang Liu, Xuan Zhu. Learning Entity and Relation Embeddings for Knowledge Graph Completion. The 29th AAAI Conference on Artificial Intelligence (AAAI'15).
Y. Yang, C. Huang, L. Xia, and C. Li. Proceedings of the 45th international ACM SIGIR conference on research and development in information retrieval, page 1434--1443. (2022)