Alterando as probabilidades a priori
Como mencionado no vídeo, você também pode alterar as probabilidades a priori para obter uma árvore de decisão. Essa é uma forma indireta de ajustar a importância das classificações incorretas para cada classe. Você pode especificar outro argumento dentro de rpart() para incluir as probabilidades a priori. O argumento que você precisa tem o seguinte formato
parms = list(prior=c(non_default_proportion, default_proportion))
O pacote rpart já está carregado no seu ambiente de trabalho.
Este exercício faz parte do curso
Modelagem de Risco de Crédito em R
Instruções do exercício
- Altere o código fornecido para construir uma árvore de decisão, incluindo o argumento
parmse mudando a proporção de não inadimplentes para 0.7 e de inadimplentes para 0.3 (elas devem sempre somar 1). Além disso, inclua tambémcontrol = rpart.control(cp = 0.001). - Faça o gráfico da árvore de decisão usando a função plot e o nome do objeto da árvore. Adicione um segundo argumento "uniform=TRUE" para obter ramificações de tamanho igual.
- Adicione rótulos à árvore usando a função
text()e o nome do objeto da árvore de decisão.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# Change the code below such that a tree is constructed with adjusted prior probabilities.
tree_prior <- rpart(loan_status ~ ., method = "class",
data = training_set)
# Plot the decision tree
# Add labels to the decision tree