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).
A. Dargahi Nobari, N. Reshadatmand, и M. Neshati. Proceedings of the 2017 ACM on Conference on Information and Knowledge Management, стр. 2035–2038. New York, NY, USA, Association for Computing Machinery, (2017)