溫馨提示×

r語言做決策樹代碼怎么寫

小億
100
2024-01-12 15:48:32
欄目: 編程語言

在R語言中,可以使用rpart庫來構(gòu)建決策樹模型。下面是一個簡單的決策樹構(gòu)建的示例代碼:

# 導(dǎo)入rpart庫
library(rpart)

# 導(dǎo)入數(shù)據(jù)集
data(iris)

# 創(chuàng)建決策樹模型
model <- rpart(Species ~ ., data = iris)

# 打印決策樹模型
print(model)

這段代碼使用了經(jīng)典的鳶尾花數(shù)據(jù)集進(jìn)行了決策樹的構(gòu)建。在構(gòu)建模型時,使用了rpart函數(shù),其中Species是目標(biāo)變量,而.表示使用所有其他變量作為預(yù)測變量。最后,使用print函數(shù)打印出構(gòu)建好的決策樹模型。

你可以根據(jù)自己的數(shù)據(jù)集和需求,進(jìn)行相應(yīng)的調(diào)整和修改。

0