準備

library(visNetwork)  # ライブラリの読み込み
library(rpart)  

決定木

res <- rpart(Species~., data=iris)
visTree(res, main = "Iris classification Tree", width = "100%")
z <- data.frame(Titanic)
titanic.data <- data.frame(
  Class = rep(z$Class, z$Freq),
  Sex = rep(z$Sex, z$Freq),
  Age = rep(z$Age, z$Freq),
  Survived = rep(z$Survived, z$Freq)
)

res <- rpart(Survived~., data = titanic.data)
visTree(res, edgesFontSize = 14, nodesFontSize = 16, width = "100%")

回帰木

res <- rpart(Petal.Length~., data=iris)
visTree(res, edgesFontSize = 14, nodesFontSize = 16, width = "100%")

参照


「データから価値を創造する」一般社団法人データマーケティングラボラトリー

Copyright© DML All Rights Reserved.