MovieLens1M基于深度学习的电影推荐系统
使用MovieLens1M数据集(数据可以从下载),实现自动编码器(AE),可变自动编码器(VAE),BERT提取电影名特征3种方法,对评分矩阵进行耦合,继而对用户做出推荐。
代码建议在Google Colab环境下运行,代码中的目录请根据自己的实际目录进行修改。
本代码主目录和子目录如下:
/ content / drive / Movie_lens / --------- ml-1m(包含数据集的文件夹) ---------自动编码器.ipynb ---------基于BERT的recommender.ipynb
1个型号:
1.1自动编码器
1.2可变自动编码器
1.3基于BERT
2实验结果:
2.1自动编码器的训练损失和验证损失的MSE
2.2变分自动编码器的训练损失和验证损失的MSE
2.3基于BERT的训练损失和测试损失的M
2021-03-25 01:32:20
6.57MB
系统开源
1