Hi all,
Suppose I have a multiclass sentence classification problem where each sentence belongs only to one of the 10 classes [0, 1, …, 9]
Can someone tell me which loss function to use? (I will use LSTMs)
Hi all,
Suppose I have a multiclass sentence classification problem where each sentence belongs only to one of the 10 classes [0, 1, …, 9]
Can someone tell me which loss function to use? (I will use LSTMs)
https://pytorch.org/docs/stable/nn.html#torch.nn.CrossEntropyLoss
It seems helpful according to the description of the doc.