How to train a model on more than one label
GabrielGhe opened this issue · 1 comments
GabrielGhe commented
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?
GabrielGhe commented
I figured out that you can have/return multiple fc layers in the forward pass.