def _predict(self, x, y):
    (_, y_) = self.solver.model(x)
    if (not self.solver.multiclass):
        y_ = y_.squeeze()
        y = y.float()
    return (y_, y)
def _predict(self, x, y):
    (_, y_) = self.solver.model(x)
    if (not self.solver.multiclass):
        y_ = y_.squeeze()
        y = y.float()
    return (y_, y)