r - 使用 caret 包运行带有 controls = cforest_unbiased() 的 cforest

Question

我想使用 caret 包运行一个公正的 cforest。这可能吗？

tc <- trainControl(method="cv",
               number=f,
               index=indexList,
               savePredictions=T,
               classProbs = TRUE,
               summaryFunction = twoClassSummary)
createCfGrid <- function(len, data) {
    g = createGrid("cforest", len, data)
    g = expand.grid(.controls = cforest_unbiased(mtry = 5, ntree = 1000))
    return(g)
}
set.seed(1)
(cfMatFit <- train(as.factor(f1win) ~ .,
                   data=df,
                   method="cforest",
                   metric="ROC",
                   trControl=tc,
                   tuneGrid = createCfGrid))

错误是Error in as.character.default(<S4 object of class "ForestControl">) : no method for coercing this S4 class to a vector

这是因为 cforest_control() 无法强制转换为数据框。如果我使用，该功能确实有效：

...
g = expand.grid(.mtry = 5)
...

但是，如果我想更改 ntree，这没有效果：

...
g = expand.grid(.mtry = 5, .ntree = 1000)
...

这不会像 randomForest 那样出错。

Answer 1

网格应该是一个简单的数据框，其中有一列名为.mtry. 编码

 g = createGrid("cforest", len, data)

将为您生成。如果要指定ntree，只需将controls对象作为另一个参数传递给train但省略mtry：

 mod <- train(Species ~ ., data = iris,
              method = "cforest",
              controls = cforest_unbiased(ntree = 10))

caret照顾mtry你的变化。

最大限度