R语言tree()函数的使用代码
时间: 2024-05-24 17:14:00 浏览: 121
r-tree source code
5星 · 资源好评率100%
生成决策树的步骤如下:
1.加载R语言中的rpart包和rpart.plot包。
library(rpart)
library(rpart.plot)
2.准备数据将数据导入R语言中,使用以下:
data(iris)
head(iris)
3.使用决策树算法生成决策树模型,使用以下代码:
tree <- rpart(Species ~ ., data = iris, method = "class")
其中,Species是目标变量,.代表所有其他变量,data表示数据集,method表示使用分类算法生成模型。
4.可视化决策树,使用以下代码:
rpart.plot(tree, type = 4, extra = 101)
其中,type = 4 表示使用树形结构展示决策树,extra = 101 是调整图像大小的参数。
以上就是R语言tree()函数的使用代码。
阅读全文