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

Posted 2022-03-25 炫云云

文章目录

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

FM模型

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

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

FM模型计算公式：

y ^