DIN、DIEN 模型

本仓库Fork自yongqyu/DIEN-tf2，为其中BASE模型、DIN模型、DIEN模型的核心代码部分加上了注释。 如果想要了解这两个模型的理论知识，可以参考我的两篇博客：

1. CTR深度学习模型之 DIN(Deep Interest Network) 的理解与例子
2. CTR深度学习模型之 DIEN(Deep Interest Evolution Network) 的理解与示例

或者参考论文原文:

1. Deep Interest Network for Click-Through Rate Prediction
2. Deep Interest Evolution Network for Click-Through Rate Prediction

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