code: def _preprocess_Y(self, Y, k):
          
          Y = Y.clone()
  
          # If preds, convert to probs
          if Y.dim() == 1 or Y.shape[1] == 1:
              Y = pred_to_prob(Y.long(), k=k)
          return Y
docs: Convert Y to prob labels if necessary