How to remedy unbalanced classes?

Not sure the best way to correct my class balance issue. I don’t think I should remove images of my over represented class but I’m not sure what else to do…

Most models work better if your classes are fairly evenly balanced.

Common strategies to mitigate unbalanced classes are collecting more data or oversampling the under-represented classes.

A few things you can do would be to use synthetic augmentation, do some undersampling (or oversampling!), etc.

Here’s a really good breakdown to explain in more detail.

1 Like