Monday, 27 April 2015

How to perform multi-label learning with LSTM using theano?

I have some text data with multiple labels for each document. I want to train a LSTM network using Theano for this dataset. I came across http://deeplearning.net/tutorial/lstm.html but it only facilitates a binary classification task. If anyone has any suggestions on which method to proceed with, that will be great. I just need an initial feasible direction, I can work on.

You can change the last layer of the model. It would have a vector of target where each element is 0 or 1, depending if you have the target or not.

No comments :

Post a Comment