bentrevett/pytorch-sentiment-analysis

How to train a model on more than one label

GabrielGhe opened this issue · 1 comments

Given the multi-class text analysis notebook, how would that work with a two-dimensional output layer?

input = { text: ['word1', 'word2'], category: 'article', sentiment: 'good' }

The input is the text and the labels are category and sentiment.
How would that work?

I figured out that you can have/return multiple fc layers in the forward pass.