Question

这是关于贝叶斯优化实施的this answer。我无法理解以下定义函数xgb.cv.bayes（）的R代码。代码如下：

xgb.cv.bayes <- function(max.depth, min_child_weight, subsample, colsample_bytree, gamma){
  cv <- xgv.cv(params = list(booster = 'gbtree', eta = 0.05,
                         max_depth = max.depth,
                         min_child_weight = min_child_weight,
                         subsample = subsample,
                         colsample_bytree = colsample_bytree,
                         gamma = gamma,
                         lambda = 1, alpha = 0,
                         objective = 'binary:logistic',
                         eval_metric = 'auc'),
             data = data.matrix(df.train[,-target.var]),
             label = as.matrix(df.train[, target.var]),
             nround = 500, folds = cv_folds, prediction = TRUE,
             showsd = TRUE, early.stop.round = 5, maximize = TRUE,
             verbose = 0
  )
  list(Score = cv$dt[, max(test.auc.mean)],
   Pred = cv$pred)
}

我无法理解在关闭xgb.cv（）括号后出现的以下代码部分：

list(Score = cv$dt[, max(test.auc.mean)],
   Pred = cv$pred)

或者非常简短，我不理解以下语法：

xgb.cv.bayes <- function(max.depth, min_child_weight, subsample, colsample_bytree, gamma){
  cv <- xgv.cv(...)list(...)
}

我将非常感激理解这个R语法，在哪里可以找到更多这方面的例子。

Answer 1

在R中，函数中最后一个表达式的值自动为该函数的返回值。因此，您提供的功能只有两个步骤：

计算xgv.cv(...)的结果并将结果存储在a中变量cv
创建一个包含两个条目的列表（Score和Pred）其值从cv。

由于创建列表的表达式是函数中的最后一个表达式，因此列表自动为返回值。因此，如果您执行test <- xgb.cv.bayes(...)，则可以访问test$Score和test$Pred。这是否回答了你的问题？

了解贝叶斯优化代码中的R语法

1 个答案: