Att sköta klassificeringsträd

Övervakat lärande i R: Klassificering

Brett Lantz

Instructor

För-beskärning

Tidig stopp med MaxDepth

Tidig stopp med MinSplit

Övervakat lärande i R: Klassificering

Efter-beskärning

Beskärning av en komplex gren

Graf över fel vs. komplexitet

Övervakat lärande i R: Klassificering

För- och efter-beskärning med R

# pre-pruning with rpart
library(rpart)
prune_control <- rpart.control(maxdepth = 30, minsplit = 20)

m <- rpart(repaid ~ credit_score + request_amt,
           data = loans,
           method = "class",
           control = prune_control)
# post-pruning with rpart
m <- rpart(repaid ~ credit_score + request_amt,
           data = loans,
           method = "class")

plotcp(m)

m_pruned <- prune(m, cp = 0.20)
Övervakat lärande i R: Klassificering

Nu kör vi en övning!

Övervakat lärande i R: Klassificering

Preparing Video For Download...