Update reader.py

Author: tanzhongyi

lesson9/reader.py

@@ -1,10 +1,9 @@
 #!/usr/bin/env python2.7
 # -*- encoding:utf-8 -*-
 """
-数据集读取脚本
+数据集读取脚本
 Created on 2017-12-2
-@author: PaddlePaddle CTR Model
-@copyright: www.baidu.com
+
 """
 from utils import logger, TaskMode, load_dnn_input_record, load_lr_input_record
 
@@ -14,31 +13,31 @@
 class Dataset(object):
  def train(self, path):
  '''
- 载入数据集
+ 载入数据集
  '''
  logger.info("load trainset from %s" % path)
  mode = TaskMode.create_train()
  return self._parse_creator(path, mode)
 
  def test(self, path):
  '''
- 载入测试集
+ 载入测试集
  '''
  logger.info("load testset from %s" % path)
  mode = TaskMode.create_test()
  return self._parse_creator(path, mode)
 
  def infer(self, path):
  '''
- 载入预测集
+ 载入预测集
  '''
  logger.info("load inferset from %s" % path)
  mode = TaskMode.create_infer()
  return self._parse_creator(path, mode)
 
  def _parse_creator(self, path, mode):
  '''
- 稀疏化数据集
+ 稀疏化数据集
  '''
 
  def _parse():
@@ -58,7 +57,7 @@ def _parse():
 
 def load_data_meta(path):
  '''
- 从指定路径中读取meta数据，返回lr模型维度和dnn模型维度
+ 从指定路径中读取meta数据，返回lr模型维度和dnn模型维度
  '''
  with open(path) as f:
  lines = f.read().split('\n')
@@ -69,4 +68,4 @@ def load_data_meta(path):
  res = map(int, [_.split(':')[1] for _ in lines])
  logger.info('dnn input dim: %d' % res[0])
  logger.info('lr input dim: %d' % res[1])
- return res
+ return res