这是一个微型推荐系统demo。这个推荐系统目前还没完善,我还没有把之前所有的代码合并进来。
目前为止,它包含8个常见的深度学习推荐模型。召回阶段包括DSSM、MIND、NeuralCF、YoutubeDNN;排序阶段包含DeepFM、DIEN、DIN、Wide & Deep.需要注意的是,召回阶段的召回结果并没有 作为排序阶段的模型输入:这是因为我所采用的Movielens数据集体量较小,没有做这一漏斗阶段的必要。
此外,这里召回和排序的数据预处理手段(特征工程)并不一样。召回阶段仅仅使用UserID和ItemID作为特征,排序阶段则做了大量的特征工程。