In this repository, we implement many recent deep learning based recommendation models with Tensorflow.
We implemented both rating estimation and top-n recommendation models.
- AutoRec (www)
- CDAE (WSDM)
- NeuMF (WWW)
- CML (WWW)
- etc.
You can run this code from Test/testItemRanking.py or Test/testRatingPred.py
- Tensorflow 1.7+
- Python 3.5+
- numpy
- scipy
- sklearn
- pandas
- Add more models
- Different Evaluation Protocals
- Code Refactor
To acknowledge use of this open source package in publications, please cite the following paper:
- Shuai Zhang, Lina Yao, and Aixin Sun. "Deep learning based recommender system: A survey and new perspectives." arXiv preprint arXiv:1707.07435 (2017).
Thank you for your support!
Contributions and issues are always welcome. You can also contact me via email: [email protected]