for example, if a dataset contains 100 positive and 300 negative examples of a single class,
for example, if a dataset contains 100 positive and 300 negative examples of a single class, | BCEWithLogitsLoss