Skip to content

Latest commit

 

History

History
36 lines (23 loc) · 905 Bytes

README.md

File metadata and controls

36 lines (23 loc) · 905 Bytes

CDAE

Implementation of Collaborative Denoising Auto-Encoder (CDAE) with the Keras.

References

  • Yao Wu, Christopher DuBois, Alice X. Zheng, Martin Ester. Collaborative Denoising Auto-Encoders for Top-N Recommender Systems. The 9th ACM International Conference on Web Search and Data Mining (WSDM'16), p153--162, 2016.
  • F. Maxwell Harper and Joseph A. Konstan. 2015. The MovieLens Datasets: History and Context. ACM Transactions on Interactive Intelligent Systems (TiiS) 5, 4, Article 19 (December 2015), 19 pages.

Usage

First, install libraries:

pip install -r requirements.txt

Then,

# CPU
python train.py

# GPU
THEANO_FLAGS=device=gpu,floatX=float32 python train.py

TODO

  • implement negative sampling
  • change the way of init

Licence

MIT License Copyright (c) 2016 Tsukasa ŌMOTO