google-research
Hyperbolic Collaborative Filtering (CF) Embedding Library
This project includes an implementation of several CF models in tensorflow 2.0, including:
Euclidean models:
- SMFactor [1]
Hyperbolic models:
- HyperML [2]
[1] A simple matrix facrotization model following Funk, Simon: https://en.wikipedia.org/wiki/Matrix_factorization_(recommender_systems)#Funk_MF
[2] HyperML: A Boosting Metric Learning Approach in Hyperbolic Space for Recommender Systems, Tran Dang Quang Vinh, Yi Tay, Shuai Zhang, Gao Cong, Xiao-Li Li, WSDM 2020.