Return the correct numpy array from boosting prediction