Implementation of Deep Learning based Recommender Algorithms with Tensorflow.
Switch branches/tags
Nothing to show
Clone or download
Latest commit ef63715 Nov 9, 2018
Permalink
Failed to load latest commit information.
config rename files Sep 15, 2018
data add neural factorization machine. Sep 15, 2018
models fix the import error. Sep 16, 2018
test fix the import error. Sep 16, 2018
utils updating the ranking label. Nov 8, 2018
LICENSE Add GPL License May 21, 2018
README.md Update README.md Nov 9, 2018

README.md

DeepRec

In this repository, we implement many recent deep learning based recommendation models with Tensorflow.

Collaborators

Shuai Zhang, Yi Tay, Bin Wu

Implemented Algorithms

We implemented both rating estimation and top-n recommendation models.

  • I-AutoRec and U-AutoRec (www'15)
  • CDAE (WSDM'16)
  • NeuMF (WWW'17)
  • CML (WWW'17)
  • LRML (WWW'18) (DRAFT ONLY, testing will come soon)
  • NFM (SIGIR'17)
  • NNMF (arxiv)
  • etc.

You can run this code from Test/testItemRanking.py or Test/testRatingPred.py

Requirements

  • Tensorflow 1.7+, Python 3.5+, numpy, scipy, sklearn, pandas

To do

  • Add more models
  • Different Evaluation Protocals
  • Code Refactor

Citation

To acknowledge use of this open source package in publications, please cite the following paper:

@article{zhang2017deeprec,
  title={Deep learning based recommender system: A survey and new perspectives},
  author={Zhang, Shuai and Yao, Lina and Sun, Aixin and Tay, Yi},
  journal={arXiv preprint arXiv:1707.07435},
  year={2017}
}

Thank you for your support!!!

Contributions and issues are always welcome. You can also contact me via email: cheungshuai@outlook.com