Class imbalance in one-hot encoding

I don’t think using the WeightedRandomSampler would work out of the box for a multi-label classification and you might want to take a look at this post for some suggestions.

1 Like