Мультиклассификация с использованием XGBClassifier - PullRequest
0 голосов
/ 21 мая 2019

Я использую XGBClassifier для мультиклассовой классификации (5 классов - [1,2,3,4,5]).Я установил целевой параметр как «multi: softmax», но все же, когда я прогнозирую, используя мою модель, я получаю непрерывные значения вместо целых.

Я также пытался указать параметр num_class, но он все еще предсказывает непрерывные значения.

model = XGBClassifier(learning_rate = 0.1,n_estimators = 200, objective='multi:softmax')

model.fit(x1, y1, eval_set=[(x1,y1),(x2, y2)], eval_metric='mlogloss')

Ожидаемый результат = [1,2,3,3,2,3,4,4,5,5,1 .... и т.д.] # целочисленные значения

фактический результат = [2,333, 1,455, 2,122, 1,76 .... и т. д.] # непрерывные значения

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...