Prevenire alberi troppo cresciuti
L'albero cresciuto sull'intero insieme di dati dei richiedenti è diventato estremamente grande e complesso, con centinaia di split e foglie che contengono solo pochi richiedenti. Questo albero sarebbe quasi impossibile da interpretare per un addetto ai prestiti.
Usando i metodi di pre-pruning per l'arresto anticipato, puoi impedire a un albero di crescere troppo in altezza e complessità. Osserva come le opzioni di controllo di rpart per la profondità massima dell'albero e il numero minimo di suddivisioni influenzano l'albero risultante.
rpart è già stato caricato.
Questo esercizio fa parte del corso
Apprendimento supervisionato in R: Classificazione
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
# 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(___)