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

问题描述 投票:3回答:1

我想使用插入包运行一个没有偏见的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那样出错。

r random-forest r-caret
1个回答
5
投票

网格应该是一个简单的数据框,其中一列名为.mtry。代码

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

会为你生成。如果你想指定ntree,你只需要将controls对象作为另一个参数传递给train,但不要使用mtry

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

caret负责为您改变mtry

马克斯

© www.soinside.com 2019 - 2024. All rights reserved.