Skip to content

Officially unofficial TensorFlow code for 'Collaborative Deep Learning for Recommender Systems' - SIGKDD

License

Notifications You must be signed in to change notification settings

js05212/CollaborativeDeepLearning-TensorFlow

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

This is the author's officially unofficial TensorFlow implementation of CDL. Note that the code is a simplified version of CDL and is used for demonstration only.

To train the model

  • pretrain model (CDL) run 'python train_sdae.py'
  • train CDL run 'python train_cdl.py'

To evaluate performance

  • run lib/mainEvaluate.m under Matlab

Other materials

Other implementations (third-party):

Acknowledgement

Reference:

Collaborative Deep Learning for Recommender Systems

@inproceedings{DBLP:conf/kdd/WangWY15,
  author    = {Hao Wang and
               Naiyan Wang and
               Dit{-}Yan Yeung},
  title     = {Collaborative Deep Learning for Recommender Systems},
  booktitle = {SIGKDD},
  pages     = {1235--1244},
  year      = {2015}
}