PMF-Pytorch
使用Pytorch的PMF的实现 ***** 这是在建议中使用Adam更新规则进行概率矩阵分解的Pytorch代码。 所有文件都井井有条,易于理解您可以使用movielen-1m来测试此代码。 请注意,此代码中的数据路径均为相对路径。 这些文件如下: ===> 1。 0.data_process-1.py 生成pmf_main.py文件的数据===> 2。 PMF_main.py pmf算法的主文件定义了一些超参数。 ===> 3 PMF_model.py 该文件包含主要的pmf模型定义。 ===> 4个评价.py 该文件定义了此算法的评估指标方法。(此文件中的RMSE) 运行说明: 0.data_process-1.py ---> PMF_main.py 论文参考: 概率矩阵分解
2021-08-23 11:51:18
896KB
Python
1