調整分類樹

R 監督式學習:分類

Brett Lantz

Instructor

前剪枝

以 MaxDepth 提前停止

以 MinSplit 提前停止

R 監督式學習:分類

後剪枝

修剪複雜分支

繪製錯誤與複雜度

R 監督式學習:分類

在 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)
R 監督式學習:分類

一起來練習吧!

R 監督式學習:分類

Preparing Video For Download...