Een Bayesiaans model plotten
In eerdere oefeningen hebben we een Bayesiaans model geschat dat de populariteit van een nummer (popularity) voorspelt op basis van de leeftijd (song_age). Laten we het model nu visualiseren. Gebruik de al geladen gegevensset songs en het object stan_model om een visualisatie te maken met de gegevens en de geschatte regressielijn met ggplot2.
Deze oefening maakt deel uit van de cursus
Bayesiaanse regressiemodellering met rstanarm
Oefeninstructies
- Sla een nette samenvatting van de modelparameters op in
tidy_coef - Haal de geschatte intercept en helling op uit
tidy_coef - Maak een plot met de gegevens en de geschatte regressielijn, met
song_ageop de x-as enpopularityop de y-as
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
# Save the model parameters
tidy_coef <- ___(stan_model)
# Extract intercept and slope
model_intercept <- tidy_coef$___[1]
model_slope <- tidy_coef$___[2]
# Create the plot
ggplot(songs, aes(x = ___, y = ___)) +
geom_point() +
geom_abline(intercept = ___, slope = ___)