Skip to content
/ VAE Public

Variational Autoencoders for Collaborative Filtering

Notifications You must be signed in to change notification settings

KV9801/VAE

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 

Repository files navigation

Variational Autoencoders for Collaborative Filtering

Download the dataset at http://files.grouplens.org/datasets/movielens/ml-20m.zip

To run the code, perform the following steps:

  • Put the source code (VAE.py or DAE.py) and ratings.csv (present in the dataset file) in the same directory
  • Run the code using the following command: CUDA_VISIBLE_DEVICES=<GPU_ID> python <DAE/VAE>.py

Following are the prerequisites to run the code:

  • CUDA
  • Tensorflow2
  • Python 3.x

About

Variational Autoencoders for Collaborative Filtering

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages