Curare gli alberi di classificazione

Apprendimento supervisionato in R: Classificazione

Brett Lantz

Instructor

Pre-pruning

Early stopping con MaxDepth

Early stopping con MinSplit

Post-pruning

Potatura di un ramo complesso

Errore vs complessità

Pre- e post-pruning con 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)

Esercitiamoci!

Apprendimento supervisionato in R: Classificazione