温馨提示×

r语言做决策树代码怎么写

小亿
92
2024-01-12 15:48:32
栏目: 编程语言

在R语言中,可以使用rpart库来构建决策树模型。下面是一个简单的决策树构建的示例代码:

# 导入rpart库
library(rpart)

# 导入数据集
data(iris)

# 创建决策树模型
model <- rpart(Species ~ ., data = iris)

# 打印决策树模型
print(model)

这段代码使用了经典的鸢尾花数据集进行了决策树的构建。在构建模型时,使用了rpart函数,其中Species是目标变量,而.表示使用所有其他变量作为预测变量。最后,使用print函数打印出构建好的决策树模型。

你可以根据自己的数据集和需求,进行相应的调整和修改。

0