Recommender System WITH PyTorch 🟢🟠🔴
PyTorch Implementation of classic Recommender System Models mainly used for self-learing&communication .
checkout for tensorflow branch
corresponding papers 🚀 RS_Papers 📖
Model
dataset
loss_func
metrics
state
LFM
ml-100k
MSELoss
MSE: 0.9031
🟢
BiasSVD
ml-100k
MSELoss
MSE: 0.8605
🟢
SVD++
ml-100k
MSELoss
MSE: 0.8493
🟢
Model
dataset
loss_func
metrics
state
FM
criteo
BCELoss
AUC: 0.6934
🟢
FFM
criteo
BCELoss
AUC: 0.6729
🟢
Model
dataset
loss_func
metrics
state
FPMC
ml-100k
sBPRLoss
Recall@10: 0.0622
🟢
SASRec
ml-100k
BCEWithLogitsLoss
NDCG@10: 0.1801
HR@10: 0.3595
🟢
Model
dataset
loss_func
metrics
state
RippleNet
ml-1m
BCELoss
AUC: 0.8838
🟢
Model
dataset
loss_func
metrics
state
NeuralCF
ml-100k
MSELoss
MSE: 0.3322
🟢
Model
dataset
loss_func
metrics
state
FNN
criteo
BCELoss
AUC: 0.6787
🟢
DeepFM
criteo
BCELoss
AUC: 0.6854
🟢
NFM
criteo
BCELoss
AUC: 0.6705
🟢
AFM
criteo
BCELoss
AUC: 0.6572
🟢
Model
dataset
loss_func
metrics
state
Deep Crossing
criteo
BCELoss
AUC: 0.7210
🟢
PNN
criteo
BCELoss
AUC: 0.6360
🟢
Wide&Deep
criteo
BCELoss
AUC: 0.7074
🟢
DCN
criteo
BCELoss
AUC: 0.7335
🟢
DIN
amazon book
BCELoss
AUC: 0.5988
🟢
DIN: It seems that the feature engineering(negative sampling) of paper used for amazon book
seems bad. I try hard but the auc of test cannot reach the 0.811
on amazon book
.
Model
dataset
loss_func
metrics
state
MMOE
census-income
BCEWithLogitsLoss
income-AUC: 0.9061
marry-AUC: 0.9637
🟢
ESMM
census-income
BCEWithLogitsLoss
income-ctr-AUC: 0.9242
ctcvr-AUC: 0.9122
🟢