How to handle imbalanced classes

The weights tensor should contain the current weigth for each sample, not only the inverse class counts, as shown in this example.

1 Like