Which weighted loss function you would suggest if one of the classes is underrepresented in the training dataset?
I could already reach good results on the multi class task(3 different classes) with nn.CrossEntropyLoss() as loss. With good results I mean that the net is in over 90% right with his class prediction. But when I check the probabilities I could see that they are pretty low. With low I mean the average probabilities is around 40-45% for the predicted class. Now I want to try weighted loss here because 1 out of the 3 classes is rly underrepresented and maybe I can fix the issue with the low probabilities like that.