Prevenção de árvores crescidas demais
A árvore criada no conjunto completo de dados de candidatos tornou-se extremamente grande e complexa, com centenas de divisões e nós de folhas contendo apenas alguns candidatos. Essa árvore seria quase impossível de ser interpretada por um agente de empréstimos.
Usando os métodos de pré-poda para a interrupção precoce, você pode evitar que uma árvore cresça muito grande e complexa. Veja como as opções de controle do rpart
para a profundidade máxima da árvore e a contagem mínima de divisões afetam a árvore resultante.
rpart
foi pré-carregado.
Este exercício faz parte do curso
Aprendizagem supervisionada em R: Classificação
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# 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(___)