我在R中使用rpart库。在函数中,我想返回一个在for循环中生成的rpart对象数组。但是,我不知道应该使用哪种数据结构来存储rpart对象。每个rpart对象都有许多值。您可以在下面看到生成rpart对象的代码:
rpart.fit <- rpart(result ~ . , data = this.data ,
subset = train.index, method= "class",
control=rpart.control(maxdepth=1))
答案 0 :(得分:0)
library(rpart)
this.data <- data.frame(result = runif(20), var = runif(20))
foo <- function(data){
aa <- list()
for(i in 1:10){
aa[[paste("model", i, sep="")]] <- rpart(result ~ var , data = this.data)
}
aa
}
bar <- foo(this.data)
plot(bar$model1)
plot(bar$model5)
编辑:更新了动态计算模型名称的功能。