Skip to content

Implementation of two pytorch based recommenders, Restricted Boltzmann Machine and Autoencoder.

Notifications You must be signed in to change notification settings

hwang018/recommenders_pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

24 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Recommenders (pytorch)

Include implementations on two types of recommenders:

  1. Restricted Boltzmann Machine based recommender (notebook example).
  2. Autoencoder based recommender.
  • common: includes utility functions shared by different recommender systems, for example, data preprocessing.
    • preprocessor.py
    • util.py
  • models: includes each model's Class:
    • encoder.py
  • dataset: datasets used in this project.
    • MovieLens dataset (used in this project)
    • BookCrossing dataset

About

Implementation of two pytorch based recommenders, Restricted Boltzmann Machine and Autoencoder.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published