mxnet-the-straight-dope
mxnet-the-straight-dope copied to clipboard
Training an RNN to classify sequences using just the last LSTM output
I did this as a demo for a particular project. I can polish it up if you want it.
You basically just run an LSTM on every sequence, take the last output for each sequence and the push that through a neural net.
There are 2 demos training on generated data and then one classifying sentences as Spanish or English. The language one is tested using entirely different novels.
Hi aidan - just got back to town after a couple weeks abroad plus some insane paper deadline - looking forward to reviewing this properly. Thanks for the PR!