Thanks for your response!
I’m also having another issue though - after applying ReLU to the output layer, I’m still getting all 0 values (as expected) as all the predictions from nn.Linear have been negative. Thus my confusion matrix has exclusively true negatives and false negatives, and no real predictions.
Is there something wrong with my model? It doesn’t look like there’s anything wrong but I’m not sure.
Thanks for your time!