Модель классификатора: можно ли присвоить классу метку -1?

Question

Я пытаюсь обучить алгоритм классификации машин для повышения градиента света (LGBMClassifier). У меня есть три класса в моих данных, и я назначил им -1, 0 и 1. Эти метки содержат значение c, поэтому я не хотел бы менять их на другие метки (например, 0, 1 и 2).

Когда я пытаюсь обучить модель, я получаю следующее сообщение об ошибке:

Label must be in [0, 3), but found -1 in label

Требуется, чтобы, если у меня было три метки, они были обозначены 0,1 и 2?

Answer 1

В соответствии с документами , вы можете указать дополнительный kwarg аргумент classes при инициализации классификатора. В частности, classes имеет

Тип: массив формы = [n_classes]

Следовательно, мы можем попробовать что-то вроде

clf = lightgbm.LGBMClassifier(..., classes=[-1, 0, 1])