使用XGBClassifier进行多类分类

时间:2019-05-21 11:07:56

标签: scikit-learn classification xgboost softmax multiclass-classification

我正在使用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.334、1.455、2.122、1.76 ....等]#连续值

0 个答案:

没有答案