【发布时间】:2015-07-29 13:41:13
【问题描述】:
我将 caret 包用于树模型。我知道插入符号使用 CV 来找到修剪树的最佳调整参数。
这是我使用的代码:
id2 <- sample(1:nrow(data),2/3*nrow(data))
#learn
app <- data[id2,]
#test
test <- data[-id2,]
ctrl<-trainControl(method="cv", number=8,classProbs=TRUE, summaryFunction=twoClassSummary)
mod0 <- train(class~., data=app,method="rpart",trControl=ctrl,metric="ROC")
plot(mod0)
plot(mod0$finalModel,uniform=TRUE,margin=.1);text(mod0$finalModel,cex=0.8)
这是我的数据:https://drive.google.com/open?id=1xrCXTLqKvGiGeo2X0Y1DvoSKvzbYFnyccLimceDIbZg
但是每次我运行代码时,我都会得到不同复杂度的树(因为 CV?),树并没有真正修剪,而是非常复杂,并且有很多终端节点。
我怎样才能得到一个不太复杂的树?
【问题讨论】:
-
提供可重现的示例和
sessionInfo的结果将有助于回答您的问题。 -
你是图书馆的作者吗?我给你发了一封电子邮件。
标签: r tree machine-learning r-caret rpart