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

<cours>Modélisation de régression bayésienne avec rstanarm</cours>

Voir le cours

Instructions de l’exercice

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 ce code d’exemple.

# 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