Update DIEN.py

wzhe06 · Mar 28, 2021 · eed64f2 · eed64f2
1 parent 1510849
commit eed64f2
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/TFRecModel/src/com/sparrowrecsys/offline/tensorflow/DIEN.py b/TFRecModel/src/com/sparrowrecsys/offline/tensorflow/DIEN.py
@@ -1,7 +1,13 @@
 '''
-diff with DIN
-1、GRU with attentional update gate (AUGRU) 
-2、auxiliary loss function with click or not click  movie(negetive sampleming)
+Author:
+    zcxia23, [email protected]
+
+Diff with DIN:
+    1、GRU with attentional update gate (AUGRU) 
+    2、auxiliary loss function with click or not click  movie(negetive sampleming)
+
+Reference:
+    [1] Zhou G ,  Mou N ,  Fan Y , et al. Deep Interest Evolution Network for Click-Through Rate Prediction[J].  2018.
 '''
 import pandas as pd
 import tensorflow as tf