我正在建造一片随机的森林.我正在try 将模型中的MSE值附加到数据帧中，同时更改mtry值.我附上了模拟数据和模拟RF(不是我的真正分析，只是希望在我的真实随机森林模型上使用该方法).
#Load the package
library(randomForest)
#Load in mock data
data("quakes")
#Create a for loop to iterate on mtry
for(i in 1:4) {
temp.model <- randomForest(mag ~ ., data = quakes, mtry = i)
}
temp.model$mse
有没有一种方法可以附加MSE为4个模型(这改变了mtry值)在一个框架？我们的目标是绘制MSE以可视化树.我试过几种不同的方法都没有用.
谢谢.
你是不是在找
suppressPackageStartupMessages(library(randomForest))
data("quakes")
mt = 1L:4L
x = as.data.frame(vapply(mt,
\(i) randomForest(mag ~ ., data = quakes, mtry = i)$mse,
numeric(500L))) |>
`colnames<-`(paste0("mtry", mt))
plot(x = mt, y = colMeans(x), type = "l", xlab = "", ylab = "MSE", xaxt = "n")
axis(1L, at = mt, labels = paste0("mtry", mt))
创建于2023-12-05年第reprex v2.0.2页