如何使用xgboost R树转储计算或进行预测?

时间:2016-05-12 17:40:15

标签: r tree logistic-regression xgboost

xgboost xgb.dump tree coefficient问题中获取提示。

我特别想知道eta = 0.1或0.01,概率计算与提供的答案有何不同?

我想使用树转储进行预测。

我的代码是

#Define train label and feature frames/matrix
y <- train_data$esc_ind
train_data = as.matrix(train_data)
trainX <- as.matrix(train_data[,-1])
param <- list("objective" = "binary:logistic",
              "eval_metric" = "logloss",
              "eta" = 0.5,
              "max_depth" = 2, 
              "colsample_bytree" = .8,
              "subsample" = 0.8, #0.75
              "alpha" = 1

)

#Train XGBoost
bst = xgboost(param=param, data = trainX, label = y, nrounds=2) 

trainX1 = data.frame(trainX)
mpg.fmap = genFMap(trainX1, "xgboost.fmap")
xgb.save(bst, "xgboost.model")
xgb.dump(bst, "xgboost.model_6.txt",with.stats = TRUE, fmap = "xgboost.fmap")

树看起来像:

booster[0]
0:[order.1<12.2496] yes=1,no=2,missing=2,gain=1359.61,cover=7215.25
    1:[access.1<0.196687] yes=3,no=4,missing=4,gain=3.19685,cover=103.25
        3:leaf=-0,cover=1
        4:leaf=0.898305,cover=102.25
    2:[team<6.46722] yes=5,no=6,missing=6,gain=753.317,cover=7112
        5:leaf=0.893333,cover=55.25
        6:leaf=-0.943396,cover=7056.75
booster[1]
0:[issu.1<6.4512] yes=1,no=2,missing=2,gain=794.308,cover=5836.81
    1:[team<3.23361] yes=3,no=4,missing=4,gain=18.6294,cover=67.9586
        3:leaf=0.609363,cover=21.4575
        4:leaf=1.28181,cover=46.5012
    2:[case<6.74709] yes=5,no=6,missing=6,gain=508.34,cover=5768.85
        5:leaf=1.15253,cover=39.2126
        6:leaf=-0.629773,cover=5729.64

当选择eta小于1时,xgboost的所有树叶得分系数是否为1?

1 个答案:

答案 0 :(得分:0)

实际上这是我以前监督过的实用。

使用上面的树结构,可以找到每个训练样例的概率。

参数列表是:

param <- list("objective" = "binary:logistic",
              "eval_metric" = "logloss",
              "eta" = 0.5,
              "max_depth" = 2, 
              "colsample_bytree" = .8,
              "subsample" = 0.8,
              "alpha" = 1)

对于叶子助推器[0]中设置的实例,叶子:0-3;概率为exp(-0)/(1 + exp(-0))。

对于助推器[0],叶子:0-3 +助推器[1],叶子:0-3;概率为exp(0+ 0.609363)/(1 + exp(0 + 0.609363))。

等等,随着迭代次数的增加。

我将这些值与R的预测概率相匹配,它们在10 ^( - 7)中有所不同,这可能是由于浮点减少了叶质量得分。

当R训练过的增强树在不同的环境中用于预测时,这个答案可以提供生产水平的解决方案。

对此的任何评论都将受到高度赞赏。