blob: b56b671c428edb5e5e86afbd9bc674e7b4bf4dd5 [file] [log] [blame] [view]
This is an example of using bidirection lstm to sort an array.
Firstly, generate data by:
cd data
python gen_data.py
Then, train the model by:
python lstm_sort.py
At last, test model by:
python infer_sort.py 234 189 785 763 231
and will output sorted seq
189
231
234
763
785