word2vec

Author: yinhaofeng

models/recall/word2vec/word2vec_reader.py

@@ -17,6 +17,7 @@
 import io
 import six
 import time
+import paddle
 from paddle.io import IterableDataset
 
 
@@ -106,8 +107,9 @@ def __iter__(self):
                                 np.array([int(context_id)]).astype('int64'))
                             np.random.seed(12345)
                             tmp = np.random.sample(self.neg_num)
-                            time.sleep(0.0001)
+                            np.random.seed(12345)
                             neg_array = self.cs.searchsorted(tmp)
+                            tmp = []
                             output.append(
                                 np.array([int(str(i))
                                           for i in neg_array]).astype('int64'))