Tracer un modèle bayésien

Dans les exercices précédents, nous avons estimé un modèle bayésien qui prédit la popularité d’un morceau (popularity) à partir de son ancienneté (song_age). Visualisons maintenant le modèle. En utilisant le jeu de données songs et l’objet stan_model déjà chargés, créez une visualisation qui affiche les données et la droite de régression estimée avec ggplot2.

Cet exercice fait partie du cours

Modélisation de régression bayésienne avec rstanarm

Afficher le cours

Instructions

Enregistrez un résumé ordonné des paramètres du modèle dans tidy_coef
Récupérez l’ordonnée à l’origine (intercept) et la pente estimées depuis tidy_coef
Créez un graphique affichant les données et la droite de régression estimée, avec song_age sur l’axe des x et popularity sur l’axe des y

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

# 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 = ___)

Modifier et exécuter le code