Skip to content

Latest commit

 

History

History
37 lines (32 loc) · 1.2 KB

README.md

File metadata and controls

37 lines (32 loc) · 1.2 KB

recsys model

Implementation for classic RecommenderSystem models with Pytorch


Model List

Model Paper
PMF(Probabilistic Matrix Factorization) [NIPS 2007] Probabilistic matrix factorization
CTR(Collaborative Topic Regression) [KDD 2011] Collaborative Topic Modeling for Recommending Scientific Articles
CVAE(Collaborative Variantional AutoEncoder) [KDD 2017] Collaborative Variational Autoencoder for Recommender Systems

Project Structure

project(e.g PMF)
│
│───main.py : to run the model example with data
│
│───model.py  : the model's implementation
│
│───load_data.py  : load data from '/data/xx'
│
└─── /data
│     │
│     └─── /data_name
│               │   
│               └─── data1.txt
│               │   
│               └─── data2.dat
│               │   
│               └─── README.md
│               │   
│               │   ...