Fix bugs of reader.py in ctr model

ctr/reader.py

@@ -4,51 +4,49 @@ feeding_index = {'dnn_input': 0, 'lr_input': 1, 'click': 2}
 
 
 class Dataset(object):
-    def __init__(self):
-        self.mode = TaskMode.create_train()
-
     def train(self, path):
         '''
         Load trainset.
         '''
         logger.info("load trainset from %s" % path)
-        self.mode = TaskMode.create_train()
-        self.path = path
-        return self._parse
+        mode = TaskMode.create_train()
+        return self._parse_creator(path, mode)
 
     def test(self, path):
         '''
         Load testset.
         '''
         logger.info("load testset from %s" % path)
-        self.path = path
-        self.mode = TaskMode.create_test()
-        return self._parse
+        mode = TaskMode.create_test()
+        return self._parse_creator(path, mode)
 
     def infer(self, path):
         '''
         Load infer set.
         '''
         logger.info("load inferset from %s" % path)
-        self.path = path
-        self.mode = TaskMode.create_infer()
-        return self._parse
+        mode = TaskMode.create_infer()
+        return self._parse_creator(path, mode)
 
-    def _parse(self):
+    def _parse_creator(self, path, mode):
         '''
         Parse dataset.
         '''
-        with open(self.path) as f:
+
+        def _parse():
+            with open(path) as f:
                 for line_id, line in enumerate(f):
                     fs = line.strip().split('\t')
                     dnn_input = load_dnn_input_record(fs[0])
                     lr_input = load_lr_input_record(fs[1])
-                if not self.mode.is_infer():
+                    if not mode.is_infer():
                         click = [int(fs[2])]
                         yield dnn_input, lr_input, click
                     else:
                         yield dnn_input, lr_input
 
+        return _parse
+
 
 def load_data_meta(path):
     '''