開始使用免費開始

避免決策樹長得過頭

使用完整申請人資料所長出的樹,規模龐大又複雜,包含上百個分裂,還有許多只包含少數申請人的葉節點。放在貸款審核實務上,這樣的樹幾乎不可能被解讀。

利用「預先剪枝(pre-pruning)」的提前停止方法,你可以避免決策樹長得過大、過複雜。來看看 rpart 的控制選項中,限制最大樹深(maximum tree depth)與最小分割數(minimum split count)如何影響最終的樹。

loans_trainloans_test 資料集已建立,並且已預先載入 rpart

本練習屬於課程

R 監督式學習:分類

檢視課程

動手互動練習

試著完成這個範例程式碼,體驗一下這個練習。

# Grow a tree with maxdepth of 6
loan_model <- ___

# Make a class prediction on the test set
loans_test$pred <- ___

# Compute the accuracy of the simpler tree
mean(___)
編輯並執行程式碼