Awesome
Deep Learning Recommender Systems
A repository to experiment building recommender systems using apache MXNet.
Repo structure
/data
directory used for storing model input data/libs
python modules used by model training script./recysys.py
trains the recommender system./results
directory used for storing model reports and results./docs
: research notes etc
Running the code
- Clone this repo and cd to root
- Download, unzip and store the ratings data:
$ mkdir data
$ wget 'http://files.grouplens.org/datasets/movielens/ml-1m.zip'
$ unzip -a ml-1m.zip
$ mv ./ml-1m/ratings.dat ./data/
$ rm -rf ml-1m
$ rm ml-1m.zip
- Train the model:
$ python recsys.py
ToDo
- Use normal data iterators with data in memory
- Use callback functions to reproduce HR@K plots