使用R语言写一个随机森林模型
时间: 2024-05-25 16:19:48 浏览: 77
R语言生成随机森林
好的,您可以使用R语言提供的randomForest包来编写随机森林模型。以下是一些示例代码,您可以根据您的需求进行修改:
# 导入数据
data(iris)
# 划分训练集和测试集
train <- sample(1:nrow(iris), round(nrow(iris) * 0.8))
test <- setdiff(1:nrow(iris), train)
# 构建模型
library(randomForest)
set.seed(123)
rf_model <- randomForest(Species ~ ., data = iris[train,])
# 预测测试集
rf_pred <- predict(rf_model, iris[test,], type = "class")
# 验证准确率
library(caret)
confusionMatrix(rf_pred, iris[test, "Species"])
阅读全文