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