如何将参数传递给插入符号汇总函数

Question

如何将参数传递给插入符号汇总函数？

例如， caret 的 twoClassSummary 函数具有以下签名： twoClassSummary(data, lev = NULL, model = NULL)

参数 lev 指定了我感兴趣的因子级别。现在我想告诉 caret 我对因子级别“false”的摘要感兴趣。就像是

  trainControl(summary=twoClassSummary(lev="false")) 

不起作用。如何将参数传递给汇总函数？谢谢！

Answer 1

我相信目前无法以直接的方式执行此操作，因为 train(...) 的附加参数不会传递给 twoClassSummary()。您可以使用技巧使其工作，并且您的想法非常接近。创建一个自定义函数，将值从插入符号传递到汇总函数，但在执行此操作时处理它们或替换它们。

trainControl(
    summaryFunction = function(...) empTwoClassSummary(..., lev = c("your_other_class","false"))

您可以执行相同的操作以将附加参数传递给自定义汇总函数，例如：

trainControl(
    summaryFunction = function(...) customTwoClassSummary(..., weights = c(0.1, 0.2), orgData = my_data)

与问题无关：在您的示例中，请确保您想要成为目标级别的级别是 中的第二个向量元素lev，因为 twoClassSummary() 在这些行中默认假设这一点：

lvls <- levels(data$obs)
data$y = as.numeric(data$obs == lev[2])