update dien

shenweichen · Mar 30, 2019 · 316ab68 · 316ab68
1 parent 44610af
commit 316ab68
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/.travis.yml b/.travis.yml
@@ -30,8 +30,8 @@ matrix:
     - env: TF_VERSION=1.7.0
     - env: TF_VERSION=1.7.1
     - env: TF_VERSION=1.8.0
-    - env: TF_VERSION=1.12.0 # too slow
-    - env: TF_VERSION=1.13.1 # too slow
+    #- env: TF_VERSION=1.12.0 # too slow
+    #- env: TF_VERSION=1.13.1 # too slow
   fast_finish: true
 
 cache: pip

diff --git a/deepctr/layers/sequence.py b/deepctr/layers/sequence.py
@@ -663,6 +663,7 @@ def build(self, input_shape):
             self.gru_cell = tf.nn.rnn_cell.GRUCell(self.num_units)
 
         # Be sure to call this somewhere!
+
         super(DynamicGRU, self).build(input_shape)
 
     def call(self, input_list):

diff --git a/deepctr/models/dien.py b/deepctr/models/dien.py
@@ -220,4 +220,5 @@ def DIEN(feature_dim_dict, seq_feature_list, embedding_size=8, hist_len_max=16,
 
     if use_negsampling:
         model.add_loss(alpha * aux_loss_1)
+    tf.keras.backend.get_session().run(tf.global_variables_initializer())
     return model