Arboles de decisión

Load libraries

library(tree)
library(dplyr)

Data set

db = read.csv("C:\\Users\\alfon\\OneDrive\\Desktop\\drug200.csv")
db$Sex <- as.factor(db$Sex)
db$BP <- as.factor(db$BP)
db$Cholesterol <- as.factor(db$Cholesterol)
db$Drug <- as.factor(db$Drug)

Entrenar el data set

train <- sample(seq(length(db$Drug)), round(length(db$Drug) * 0.7), replace = FALSE)

Creación del árbol

db.tree <- tree(Drug ~ Age + Sex + BP + Cholesterol + Na_to_K, data = db, subset = train)

Plot

plot(db.tree)
text(db.tree, pretty = 0)