Skip to content

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

Notifications You must be signed in to change notification settings

yongqyu/DIEN-tf2

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published