Hey @shaun, AFAIK, it’s exactly what you are saying, the result from the sigmoid for the test set is the probability of the positive class, imminent, class label 1, or even P(class = imminent), as the very good explanation from @KFrank points out. And since you have an imbalanced data set, you can work with the “weights” attribute from the BCEWithLogitsLoss, to give different weights for classes with fewer examples