如何在train()方法中找出袋外错误=" treebag"

时间:2017-11-07 14:57:39

标签: r r-caret

如何使用method =" treebag"从train()函数中提取出行包错误。

对照= trainControl(方法=" CV",数= 10) 袋=列车(X_train,as.factor(y_train),方法=" treebag",trControl =对照,详细= F)

遇到了一个选项' coob'在' ipred'打包以解决行李错误。请协助。

下面的步骤是找到OOB的正确方法吗? OOB =表(y_train,预测(B树$ finalModel,X_train,OOB = T)) 总和(DIAG(as.matrix(OOB)))/ nrow(X_train)

1 个答案:

答案 0 :(得分:0)

有一些内置代码可用于几种不同的模型(因此您可以使用trainControl(method = "oob")进行调整)。

请注意,您需要设置keepX选项:

> library(caret)
> 
> set.seed(422)
> dat <- twoClassSim(100)
> 
> mod <- train(Class ~ ., data = dat, method = "treebag",
+              trControl = trainControl(method = "none"),
+              # you'll need this to bass to the bagging function
+              keepX = TRUE)
> 
> tb_code <- getModelInfo("treebag")[[1]]
> tb_code$oob(mod$finalModel)
  Accuracy      Kappa AccuracySD    KappaSD 
0.72787041 0.45005686 0.08011663 0.16212862