计算XGBoost中的概率

时间:2019-10-04 14:20:30

标签: r regression logistic-regression xgboost predict

我开始在R中使用XGBoost,并试图将 binary:logistic 模型中的预测与使用自定义日志丢失函数生成的预测相匹配。我希望以下两次调用 predict 会产生相同的结果:

require(xgboost)

loglossobj <- function(preds, dtrain) {
  labels <- getinfo(dtrain, "label")
  preds <- 1/(1 + exp(-preds))
  grad <- preds - labels
  hess <- preds * (1 - preds)
  return(list(grad = grad, hess = hess))
}

data(agaricus.train, package='xgboost')
data(agaricus.test, package='xgboost')
train<-agaricus.train
test<-agaricus.test

model<-xgboost(data = train$data, label = train$label, nrounds=2,objective="binary:logistic")
preds = predict(model,test$data)
print (head(preds))

model<-xgboost(data = train$data, label = train$label, nrounds=2,objective=loglossobj, eval_metric = "error")
preds = predict(model,test$data)
x = 1 / (1+exp(-preds))
print (head(x))

自定义对数损失函数的模型输出未应用逻辑变换1 /(1 + exp(-x))。但是,如果这样做,两次调用 predict 的结果概率是不同的:

[1] 0.2582498 0.7433221 0.2582498 0.2582498 0.2576509 0.2750908


[1] 0.3076240 0.7995583 0.3076240 0.3076240 0.3079328 0.3231709

我敢肯定有一个简单的解释。有什么建议吗?

1 个答案:

答案 0 :(得分:0)

事实证明,此行为是由于初始条件引起的。当调用 binary:logistic binary:logit_raw 时,xgboost隐式假定base_score = 0.5,但在使用时,必须将 base_score 设置为0.0以复制其输出。自定义损失函数。在这里,base_score是所有实例的初始预测得分。

为说明起见,下面的R代码在所有三种情况下都会产生相同的预测:

[1] 0.1814032 0.8204284 0.1814032 0.1814032 0.1837782 0.1952717
[1] 0.1814032 0.8204284 0.1814032 0.1814032 0.1837782 0.1952717
[1] 0.1814032 0.8204284 0.1814032 0.1814032 0.1837782 0.1952717

输出

{{1}}