Added gen_data.py and modified README.md for bi-lstm-sort example (ap…

…ache#6549)

example/bi-lstm-sort/README.md

@@ -2,9 +2,13 @@ This is an example of using bidirection lstm to sort an array.
 
 Firstly, generate data by:
 
-    cd data
 	python gen_data.py
 
+Move generated txt files to data directory
+
+	mkdir data
+	mv *.txt data
+
 Then, train the model by:
 
     python lstm_sort.py

example/bi-lstm-sort/gen_data.py

@@ -0,0 +1,20 @@
+import random
+
+vocab = [str(x) for x in range(100, 1000)]
+sw_train = open("sort.train.txt", "w")
+sw_test = open("sort.test.txt", "w")
+sw_valid = open("sort.valid.txt", "w")
+
+for i in range(1000000):
+    seq = " ".join([vocab[random.randint(0, len(vocab) - 1)] for j in range(5)])
+    k = i % 50
+    if k == 0:
+        sw_test.write(seq + "\n")
+    elif k == 1:
+        sw_valid.write(seq + "\n")
+    else:
+        sw_train.write(seq + "\n")
+
+sw_train.close()
+sw_test.close()
+sw_valid.close()