> 文档中心 > 6.3 tensorflow2实现FM推荐系统——Python实战

6.3 tensorflow2实现FM推荐系统——Python实战

文章目录

  • FM模型
  • Criteo数据
    • 数据处理
    • 完整代码
  • tensorflow2实现FM
    • 构建FM模型
    • 损失函数加入正则项与截取梯度
    • 训练与预测
    • 参考

FM模型

FM模型的特点:为所有特征引入隐藏向量,是矩阵分解用隐藏向量代表用户和物品的做法的扩展

FM模型原理请看:19【推荐系统12】FM——TensorFlow2实现

FM模型计算公式:

y ^