mxnet-the-straight-dope icon indicating copy to clipboard operation
mxnet-the-straight-dope copied to clipboard

Training an RNN to classify sequences using just the last LSTM output

Open aodhan-domhnaill opened this issue 7 years ago • 1 comments

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.

aodhan-domhnaill avatar Oct 29 '17 01:10 aodhan-domhnaill

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!

zackchase avatar Nov 02 '17 06:11 zackchase