Zmiana prawdopodobieństw a priori
Jak wspomniano w filmie, możesz również zmienić prawdopodobieństwa a priori, aby uzyskać drzewo decyzyjne. To pośredni sposób na dostosowanie wagi błędnych klasyfikacji dla każdej klasy. Możesz przekazać dodatkowy argument do funkcji rpart(), aby uwzględnić prawdopodobieństwa a priori. Interesujący cię argument ma następującą postać:
parms = list(prior=c(non_default_proportion, default_proportion))
Pakiet rpart jest już wczytany do twojego środowiska pracy.
To ćwiczenie jest częścią kursu
Modelowanie ryzyka kredytowego w R
Instrukcje do ćwiczenia
- Zmodyfikuj podany kod tak, aby zbudować drzewo decyzyjne z argumentem
parms, ustawiając proporcję przypadków bez niewykonania zobowiązań na 0,7, a z niewykonaniem zobowiązań na 0,3 (ich suma powinna zawsze wynosić 1). Dodaj również argumentcontrol = rpart.control(cp = 0.001). - Zwizualizuj drzewo decyzyjne, używając funkcji plot z nazwą obiektu drzewa. Dodaj drugi argument
"uniform=TRUE", aby uzyskać gałęzie o jednakowej długości. - Dodaj etykiety do drzewa, używając funkcji
text()z nazwą obiektu drzewa decyzyjnego.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# 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