Using deep learning models for building recommendation systems.
We used the ml-1m dataset for explicit and implicit ratings.
- RBM (Restricted Boltzmann Machine)
- VAE (Variational Autoencoder)
To train an RBM model run
python -m rbm.train
To train a VAE model run
python -m vae.train