天天看點

推薦算法模型——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. 代碼實作