天天看点

推荐算法模型——DeepFM

论文链接:https://arxiv.org/abs/1804.04950v1

An End-to-End Wide & Deep Learning Framework for CTR Prediction

1. 模型框架

推荐算法模型——DeepFM
  • DeepFM能够自动学习高低阶特征,且是端到端的学习

2.主要结构

DeepFM由两部分组成:FM+Deep,他们共享相同的输入,其整个模型定义为:

推荐算法模型——DeepFM

2.1 FM

推荐算法模型——DeepFM
推荐算法模型——DeepFM

2.2 Deep

推荐算法模型——DeepFM

3. 代码实现