y is the test labels and yy is the predicted ones from the model,
        y_pred = self.predict(X)
        yy=y_pred.detach().numpy()[:,1].flatten()

y_pred has dimensions of (number of test samples, 2) where 2 is the number of classes