Merge pull request #167 from livc/nce

add reader shuffle

Author: Zhao Li

nce_cost/train.py

@@ -43,9 +43,12 @@ def event_handler(event):
  parameters.to_tar(f)
 
  trainer.train(
- paddle.batch(paddle.dataset.imikolov.train(word_dict, 5), 64),
- num_passes=1000,
- event_handler=event_handler)
+ paddle.batch(
+ paddle.reader.shuffle(
+ lambda: paddle.dataset.imikolov.train(word_dict, 5)(),
+ buf_size=1000), 64),
+ num_passes=1000,
+ event_handler=event_handler)
 
 
 if __name__ == "__main__":