R:使用MLR(或插入符号或....)调整XGBoost的参数

时间:2018-03-01 11:09:08

标签: r r-caret xgboost mlr

通过几个教程,我设法制作了一个成功使用XGBoost的脚本来预测波士顿住房数据集的分类价格。

但是,我无法使用CV成功调整模型的参数。即使在stackowerflow上从教程和帖子中尝试了几个解决方案之后。

到目前为止,我最好的结果是“hacky”,只调整一个参数:

steps <- seq(75,90,5)/100
    for(i in steps){
    .....
    }

但我看到所有这些花哨的设置,使用MLR或Caret或NMOF自动运行几个参数。但是,我还没有让任何人接触这些数据。我怀疑这是因为大多数都是为二进制分类设置的,但即使最好地解决这个问题,我也无法取得成功。我可以为您提供数百行不起作用的代码,但我认为最简单的方法是尽可能提供我的代码here并听取您如何从这里开始而不是淹没我糟糕的代码。

编辑:因为我甚至没有成功运行其他人的脚本。以下是一些其他细节:

 > packageVersion("mlr")
 ‘2.11’

 > packageVersion("xgboost")
 ‘0.6.4.1’

1 个答案:

答案 0 :(得分:1)

首先,更新mlr和其他必需的软件包。然后考虑来自mlr cheatsheet的快速入门示例:

library(mlr)
#> Loading required package: ParamHelpers
library(mlbench)
data(Soybean)

set.seed(180715)

soy = createDummyFeatures(Soybean, target = "Class")
tsk = makeClassifTask(data = soy, target = "Class")
ho = makeResampleInstance("Holdout", tsk)
tsk.train = subsetTask(tsk, ho$train.inds[[1]])
tsk.test = subsetTask(tsk, ho$test.inds[[1]])

lrn = makeLearner("classif.xgboost", nrounds=10)
#> Warning in makeParam(id = id, type = "numeric", learner.param = TRUE, lower = lower, : NA used as a default value for learner parameter missing.
#> ParamHelpers uses NA as a special value for dependent parameters.

cv = makeResampleDesc("CV", iters=5)
res = resample(lrn, tsk.train, cv, acc)
#> Resampling: cross-validation
#> Measures:             acc
#> [Resample] iter 1:    0.9010989
#> [Resample] iter 2:    0.9230769
#> [Resample] iter 3:    0.9120879
#> [Resample] iter 4:    0.9230769
#> [Resample] iter 5:    0.9450549
#> 
#> Aggregated Result: acc.test.mean=0.9208791
#> 

# Tune hyperparameters
ps = makeParamSet(makeNumericParam("eta", 0, 1),
                  makeNumericParam("lambda", 0, 200),
                  makeIntegerParam("max_depth", 1, 20)
)
tc = makeTuneControlMBO(budget = 100)
tr = tuneParams(lrn, tsk.train, cv5, acc, ps, tc)
#> [Tune] Started tuning learner classif.xgboost for parameter set:
#>              Type len Def   Constr Req Tunable Trafo
#> eta       numeric   -   -   0 to 1   -    TRUE     -
#> lambda    numeric   -   - 0 to 200   -    TRUE     -
#> max_depth integer   -   -  1 to 20   -    TRUE     -
#> With control class: TuneControlMBO
#> Imputation value: -0
#> [Tune-x] 1: eta=0.529; lambda=194; max_depth=18
#> [Tune-y] 1: acc.test.mean=0.7846154; time: 0.0 min

# /... output truncated .../

#> [Tune-x] 100: eta=0.326; lambda=0.0144; max_depth=19
#> [Tune-y] 100: acc.test.mean=0.9340659; time: 0.0 min
#> [Tune] Result: eta=0.325; lambda=0.00346; max_depth=20 : acc.test.mean=0.9450549

lrn = setHyperPars(lrn, par.vals = tr$x)

# Evaluate performance
mdl = train(lrn, tsk.train)
prd = predict(mdl, tsk.test)

# Final model
mdl = train(lrn, tsk)

备忘单中的更多解释(如果要复制代码和描述,请使用 .pptx 版本)。