younggyoseo/vae-cf-pytorch younggyoseo/vae-cf-pytorch

  • Stars
    star
    130
  • Rank 277,575 (Top 6 %)
  • Language
    Python
  • Created over 6 years ago
  • Updated over 3 years ago
Twitter logo Share LinkedIn logo Share Facebook logo Share
younggyoseo/vae-cf-pytorch
github

younggyoseo

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Variational Autoencoders for Collaborative Filtering - Implementation in PyTorch

vae-cf-pytorch

An Implementation of Variational Autoencoders for Collaborative Filtering (Liang et al. 2018) in PyTorch.

This repo gives you an implementation of VAE for Collaborative Filtering in PyTorch. It's model is quite simple but powerful so i made a success reproducing it with PyTorch. Every data preprocessing step and code follows exactly from Authors' Repo.

I implemented MultiDAE too but didn't test it so MultiDAE is excluded from main.py but its code is available in models.py

Requirements

PyTorch 0.4 & Python 3.6
Numpy
TensorboardX

Examples

python main.py --cuda for full training.

Dataset

You should execute python data.py first to download necessary data and preprocess MovieLens-20M dataset.

ml-20m.zip Download

Results

More Repositories

1

Ape-X

PyTorch Implementation of Distributed Prioritized Experience Replay(Ape-X)
Python
149
star
2

MWM

Masked World Models for Visual Control
Python
113
star
3

apv

Python
69
star
4

RE3

RE3: State Entropy Maximization with Random Encoders for Efficient Exploration
Jupyter Notebook
65
star
5

CaDM

CaDM: Context-aware Dynamics Model for Generalization in Model-based Reinforcement Learning
Python
63
star
6

MV-MWM

Python
51
star
7

pytorch-nfsp

Implementation of Deep Reinforcement Learning from Self-Play in Imperfect-Information Games (Heinrich and Silver, 2016)
Python
43
star
8

trajectory_mcl

Trajectory-wise Multiple Choice Learning for Dynamics Generalization in Reinforcement Learning (NeurIPS 2020)
Python
39
star
9

CQN

Coarse-to-fine Q-Network
Python
23
star
10

lasertag-v0

Implementation of Deepmind's LaserTag-v0 game in A Unified Game-Theoretic Approach to Multiagent Reinforcement Learning(2017)
Python
18
star
11

pytorch-acer

PyTorch implementation of Sample Efficient Actor-Critic with Experience Replay(ACER)
Python
16
star
12

rnn-auxiliary-loss

Learning Longer-term Dependencies in RNNs with Auxiliary Losses - Implementation in PyTorch.
Python
13
star
13

TF_DeepNLP

TensorFlow Models for NLP
Python
1
star
14

knk-c-programming

My Solutions to exercises and programming projects to K.N.King's C programming.
C
1
star