🗒️音乐推荐系统2

type
status
date
slug
summary
tags
category
icon
password

前言

在更改为使用FM之前,首先要了解为什么要用FM,以及一个开源的以torch为基础的FM模型
这里推荐两个知乎页
以及torch-fm项目pytorch-fm
值得注意的是,实际安装下来的项目里,没有hofm工程,所以运行测试示例的时候,请把和hofm相关的选择注释

运行示例代码

实际测试了一下Movieslens数据集,该工程已做好了1M和20M数据集的预处理
可以去数据集官网下
分别为:
放在示例代码同目录下后,就可以运行下方代码了:
记得在同目录下创建名为chkpt的目录存放checkpoints
 
音乐推荐系统1音乐推荐系统3
Loading...