r语言随机森林做预测
时间: 2023-09-07 21:18:26 浏览: 86
R语言随机森林预测模型
5星 · 资源好评率100%
你可以使用R语言中的随机森林算法来进行预测。下面是一个示例代码,可以帮助你使用随机森林进行预测:
```R
# 导入randomForest库
library(randomForest)
# 读取数据集
data <- read.csv("your_dataset.csv")
# 将数据集分为训练集和测试集
trainIndex <- sample(1:nrow(data), 0.7*nrow(data))
trainData <- data[trainIndex, ]
testData <- data[-trainIndex, ]
# 创建随机森林模型
rfModel <- randomForest(target_variable ~ ., data=trainData, ntree=100)
# 预测测试集
predictions <- predict(rfModel, testData)
# 计算预测准确率
accuracy <- sum(predictions == testData$target_variable) / length(predictions)
print(paste("预测准确率:", accuracy))
```
请确保将`your_dataset.csv`替换为你自己的数据集文件名,并且`target_variable`是你要预测的目标变量的列名。你还可以根据需要调整其他参数,如`ntree`来设置随机森林中树的数量。
阅读全文