I have a NN, you can see the picture below. I have small number of data (66). I split them into (44 train, 22 test). I’ve used binary_cross_entropy, After training the AUC is 0.5. I’ve tried other loss functions, but nothing changed.
The dataset is tiny and I don’t know which layers your model contains.
Could you post the model definition here so that we could check for obvious issues?
Your model won’t work, as you are hitting a shape mismatch between self.nn and self.final.
The out_features of one linear layer are used as the in_features of the next one (in the common use case), so remove the squeeze in the forward and use: