/DIN-pytorch

PyTorch Implementation of Deep Interest Network for Click-Through Rate Prediction

Primary LanguagePython

PyTorch Implementation of Deep Interest Network for Click-Through Rate Prediction

Overview

This repository contains code unofficially reimplemented the original paper KDD 2018 Deep Interest Network for Click-Through Rate Prediction in PyTorch version

To-Dos

  • Simple Architecture Reimplementation
  • Dice Activation / PReLU
  • Data Evaluation (not trained yet)
  • Dropout
  • Regularization in DiFacto
  • Mini-Batch Aware (MBA) regularization

Acknowledgements

Part of codes are adapted from the following projects