Data not balanced

[J_Johnson]

Here is how you can create a weighted vector for scaling the loss of various classes to their representation in the training dataset:

How do I get the accuracy of an unbalanced dataset or segmentation task?

Here is a function for calculating the weight vector used by CrossEntropyLoss based on a set of targets with integers. def scaled_weights(targets): ''' Gets class weights :param targets: pass in the matrix of size (batch, classes) containing correct targets or labels as one_hot :return: returns a vector for weight rescaling for use in cross entropy loss function ''' N, C = targets.size(0), targets.size(1) # get vector of total for each class totals = ta…

That gets passed into PyTorch Cross Entropy Loss weight argument.