A Pytorch implementation of Collaborative Metric Learning (CML):
Cheng-Kang Hsieh, Longqi Yang, Yin Cui, Tsung-Yi Lin, Serge Belongie, and Deborah Estrin. 2017. Collaborative Metric Learning. In Proceedings of the 26th International Conference on World Wide Web (WWW '17)
Translated from a tensorflow version
Little difference on BookCX dataset!
However, codes are tested only with id information involved. Errs may appear when other features are in need.