该项目用PyTorch框架与经典的电影推荐系统数据集MovieLens 100K复现包括DIEN、DIN、AFM、NFM、DeepFM、ALS等在内的多个模型。同时,在处理数据集的过程中,该项目将user对电影评分rating大于等于4者的label定义为1,其余为0
-
Notifications
You must be signed in to change notification settings - Fork 1
vegetable-most/recommend-system-based-on-PyTorch
About
该项目使用经典的电影推荐数据集ml-100k并基于PyTorch实现《深度学习推荐系统(王喆著)》的部分经典模型
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published