Manage corner case with empty column of labels in predict