Merge pull request #13 from gogasca/patch-1

Replace map and lambdas with list comprehensions in test file

Author: dongjun-Lee

test.py

@@ -11,7 +11,7 @@
 word_dict, reversed_dict, article_max_len, summary_max_len = build_dict("valid", args.toy)
 print("Loading validation dataset...")
 valid_x = build_dataset("valid", word_dict, article_max_len, summary_max_len, args.toy)
-valid_x_len = list(map(lambda x: len([y for y in x if y != 0]), valid_x))
+valid_x_len = [len([y for y in x if y != 0]) for x in valid_x]
 
 with tf.Session() as sess:
     print("Loading saved model...")
@@ -24,7 +24,7 @@
 
     print("Writing summaries to 'result.txt'...")
     for batch_x, _ in batches:
-        batch_x_len = list(map(lambda x: len([y for y in x if y != 0]), batch_x))
+        batch_x_len = [len([y for y in x if y != 0]) for x in batch_x]
 
         valid_feed_dict = {
             model.batch_size: len(batch_x),
@@ -33,7 +33,7 @@
         }
 
         prediction = sess.run(model.prediction, feed_dict=valid_feed_dict)
-        prediction_output = list(map(lambda x: [reversed_dict[y] for y in x], prediction[:, 0, :]))
+        prediction_output = [[reversed_dict[y] for y in x] for x in prediction[:, 0, :]]
 
         with open("result.txt", "a") as f:
             for line in prediction_output: