fix word2vec reader

Author: wangzhen38

models/rank/dlrm/criteo_reader.py

@@ -14,7 +14,7 @@
 
 from __future__ import print_function
 import numpy as np
-
+import paddle
 from paddle.io import IterableDataset
 
 

models/recall/word2vec/word2vec_infer_reader.py

@@ -25,13 +25,12 @@ def __init__(self, file_list, config):
         super(RecDataset, self).__init__()
         self.file_list = file_list
         self.config = config
+        self.config_abs_dir = config.get("config_abs_dir", None)
         self.init()
 
     def init(self):
         dict_path = self.config.get("runner.word_id_dict_path")
-        pwd = str(os.getcwd())
-        if pwd[-8:] != 'word2vec':
-            dict_path = os.path.join(pwd, 'models/recall/word2vec', dict_path)
+        dict_path = os.path.join(self.config_abs_dir, dict_path)
         self.word_to_id = dict()
         self.id_to_word = dict()
         with io.open(dict_path, 'r', encoding='utf-8') as f:

models/recall/word2vec/word2vec_reader.py

@@ -46,13 +46,12 @@ def __init__(self, file_list, config):
         super(RecDataset, self).__init__()
         self.file_list = file_list
         self.config = config
+        self.config_abs_dir = config.get("config_abs_dir", None)
         self.init()
 
     def init(self):
         dict_path = self.config.get("runner.word_count_dict_path")
-        pwd = str(os.getcwd())
-        if pwd[-8:] != 'word2vec':
-            dict_path = os.path.join(pwd, 'models/recall/word2vec', dict_path)
+        dict_path = os.path.join(self.config_abs_dir, dict_path)
         self.window_size = self.config.get("hyper_parameters.window_size")
         self.neg_num = self.config.get("hyper_parameters.neg_num")
         self.with_shuffle_batch = self.config.get(