Fixing the problem where the label is expected to be a non-negative integer for regression.