Skip to content

rat-nick/deepRec

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

deepRec

Using deep learning models for building recommendation systems.

Dataset

We used the ml-1m dataset for explicit and implicit ratings.

Models

  • RBM (Restricted Boltzmann Machine)
  • VAE (Variational Autoencoder)

Usage

Training models

To train an RBM model run

python -m rbm.train

To train a VAE model run

python -m vae.train

Releases

No releases published

Packages

No packages published