Merge pull request #477 from wanghaoshuang/fix_text_classification

fix data reader error of text classification.
PaddlePaddle · Nov 16, 2017 · f980143 · f980143
2 parents 950f451 + 10ee066
commit f980143
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/text_classification/train.py b/text_classification/train.py
@@ -46,10 +46,10 @@ def train(topology,
         word_dict = paddle.dataset.imdb.word_dict()
         train_reader = paddle.batch(
             paddle.reader.shuffle(
-                lambda: paddle.dataset.imdb.train(word_dict), buf_size=1000),
+                lambda: paddle.dataset.imdb.train(word_dict)(), buf_size=1000),
             batch_size=100)
         test_reader = paddle.batch(
-            lambda: paddle.dataset.imdb.test(word_dict), batch_size=100)
+            lambda: paddle.dataset.imdb.test(word_dict)(), batch_size=100)
 
         class_num = 2
     else: