Movielens1M movie recommendation system:使用MovieLens数据集实现了基于自动编码器(AE)可变自动编码器(VAE)
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
下载地址
用户评论