Predicția supraviețuirii vrabielor

În acest exercițiu, vei prezice probabilitatea de supraviețuire folosind modelul de supraviețuire a vrabielor construit în exercițiul anterior.

Reține că, atunci când apelezi predict() (docs) pentru a obține probabilitățile prezise dintr-un model glm(), trebuie să specifici că dorești răspunsul:

predict(model, type = "response")

Altfel, predict() aplicat pe un model de regresie logistică returnează log-odds prezise ale evenimentului, nu probabilitatea.

Vei folosi și funcția GainCurvePlot() (docs) pentru a reprezenta grafic curba de câștig a predicțiilor modelului. Dacă această curbă este apropiată de curba ideală („wizard"), înseamnă că modelul a sortat bine vrabiurile: cu alte cuvinte, modelul a prezis că vrabiurile care au supraviețuit efectiv ar avea o probabilitate mai mare de supraviețuire. Argumentele funcției GainCurvePlot() sunt:

frame: cadrul de date cu coloana de predicții și coloana cu valorile reale
xvar: numele coloanei de predicții (ca șir de caractere)
truthVar: numele coloanei cu rezultatul real (ca șir de caractere)
title: un titlu pentru grafic (ca șir de caractere)

GainCurvePlot(frame, xvar, truthVar, title)

Cadrul de date sparrow și modelul sparrow_model au fost preîncărcate.

Acest exercițiu face parte din cursul

Învățare supervizată în R: Regresia

Vezi cursul

Instrucțiuni pentru exercițiu

Creează o nouă coloană în sparrow, numită pred, care să conțină predicțiile pe datele de antrenament.
Apelează GainCurvePlot() pentru a genera curba de câștig a predicțiilor. Modelul sortează bine vrabiurile în funcție de dacă au supraviețuit sau nu?

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

# sparrow is available
summary(sparrow)

# sparrow_model is available
summary(sparrow_model)

# Make predictions
sparrow$pred <- ___

# Look at gain curve
___(___, ___, ___, "sparrow survival model")

Editează și rulează codul