XGBoost预测

Question

我使用XGBoost为24个类做了一个多类分类任务.我现正以下列方式训练该模式：

param = {'max_depth': max_depth, 'eta': learning_rate, 'silent': 1, 'objective': 'multi:softmax', 'num_class': 24}
bst = xgb.train(param, dtrain, num_round)
bst = xgb.Booster({'nthread': 4})
predictions = bst.predict(dval)

我得到的预测是这样的：

[ 0.5  0.5  0.5 ...,  0.5  0.5  0.5]

我不知道为什么它预测所有的类都是0.5，而不是0到23中的一个指数。我需要使用其他的客观参数吗？

Answer 1

我不知道python，但是我想用下面的行用一个新创建的助推对象覆盖经过训练的对象(0.5是默认的预测)

bst = xgb.Booster({'nthread': 4})