There are 15 nodes in my output layer, one for each class.
I’m using CrossEntropyLoss to calculate the loss.
How do I calculate the accuracy here, the number of predicted labels that are equal to the target labels?
When I was using softmax and not CrossEntropyLoss I would just do target == torch.argmax(prediction) but since CrossEntropyLoss applies softmax itself, I’m not using softmax on final layer.
Would really appreciate if someone could help me out, thanks!