xiaoketongxue / DIEN-tf2

Deep Interest Network for Click-Through Rate Prediction / Deep Interest Evolution Network for Click-Through Rate Prediction

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Deep Interest Network for Click-Through Rate Prediction
Deep Interest Evolution Network for Click-Through Rate Prediction

I reference zhougr1993 and mouna99 code and converte it to TensorFlow 2.0.
This code performs similarly to the paper on ml-20 and amazon datasets.
You can modify the model called in main.py and then utilize a model such as Base, DIN, DIEN.

Requirements

  • python 3.6
  • tensorflow 2.0

Run python main.py

About

Deep Interest Network for Click-Through Rate Prediction / Deep Interest Evolution Network for Click-Through Rate Prediction


Languages

Language:Jupyter Notebook 75.4%Language:Python 24.6%