用R语言实现随机森林
时间: 2023-07-20 10:09:09 浏览: 93
好的,我可以为您解答。在 R 语言中,可以使用 randomForest 包来实现随机森林算法。
首先,需要安装 randomForest 包:
```R
install.packages("randomForest")
```
接下来,可以使用 randomForest 函数来训练模型。以下是一个简单的例子:
```R
library(randomForest)
# 读入数据
data(iris)
# 将数据集分为训练集和测试集
trainIndex <- sample(1:nrow(iris), 100)
trainData <- iris[trainIndex, ]
testData <- iris[-trainIndex, ]
# 训练模型
model <- randomForest(Species ~ ., data = trainData, ntree = 100)
# 预测测试集
pred <- predict(model, testData)
# 计算准确率
accuracy <- sum(pred == testData$Species) / nrow(testData)
cat("Accuracy:", accuracy)
```
在上述代码中,我们使用 iris 数据集作为例子,将数据集分为训练集和测试集,然后使用 randomForest 函数训练模型。在训练模型时,我们将 Species 列作为目标变量,其他列作为特征变量。我们设置 ntree 参数为 100,表示使用 100 棵决策树来构建随机森林。最后,我们使用预测函数 predict 来预测测试集,然后计算准确率。
希望以上内容能够帮到您。如果您有任何问题,请随时提出。
阅读全文