fix bug

Author: yinhaofeng

models/match/multiview-simnet/dygraph_model.py

@@ -27,7 +27,7 @@ def create_model(self, config):
         query_encode_dim = config.get('hyper_parameters.query_encode_dim', 128)
         title_encode_dim = config.get('hyper_parameters.title_encode_dim', 128)
         emb_size = config.get('hyper_parameters.sparse_feature_dim', 6327)
-        emb_dim = config.get('hyper_parameters.emb_dim', 128)
+        emb_dim = config.get('hyper_parameters.embedding_dim', 128)
         hidden_size = config.get('hyper_parameters.hidden_size', 128)
         margin = config.get('hyper_parameters.margin', 0.1)
         query_len = config.get('hyper_parameters.query_len', 79)

models/recall/mind/mind_reader.py

@@ -51,6 +51,7 @@ def init(self):
         self.items = list(self.items)
 
     def __iter__(self):
+        random.seed(12345)
         while True:
             user_id_list = random.sample(self.users, self.batch_size)
             if self.count >= self.batches_per_epoch * self.batch_size:

models/recall/word2vec/word2vec_reader.py

@@ -23,12 +23,14 @@
 class NumpyRandomInt(object):
     def __init__(self, a, b, buf_size=1000):
         self.idx = 0
+        np.random.seed(12345)
         self.buffer = np.random.randint(a, b, buf_size)
         self.a = a
         self.b = b
 
     def __call__(self):
         if self.idx == len(self.buffer):
+            np.random.seed(12345)
             self.buffer = np.random.randint(self.a, self.b, len(self.buffer))
             self.idx = 0
 
@@ -100,7 +102,7 @@ def __iter__(self):
                                 np.array([int(target_id)]).astype('int64'))
                             output.append(
                                 np.array([int(context_id)]).astype('int64'))
-
+                            np.random.seed(12345)
                             neg_array = self.cs.searchsorted(
                                 np.random.sample(self.neg_num))
                             output.append(