Implementation for classic RecommenderSystem models with Pytorch
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(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
│ │
│ │ ...