1. Apprendre
    2. /
  2. Cours
    3. /
  3. Apprentissage supervisé avec scikit-learn
Connected

Exercice

Ajuster et prédire en régression

Vous avez maintenant vu le fonctionnement de la régression linéaire. Votre tâche est de créer un modèle de régression linéaire multiple en utilisant toutes les variables explicatives du jeu de données sales_df, qui a été préchargé pour vous. Pour rappel, voici les deux premières lignes :

     tv        radio      social_media    sales
1    13000.0   9237.76    2409.57         46677.90
2    41000.0   15886.45   2913.41         150177.83

Vous utiliserez ensuite ce modèle pour prédire les ventes à partir des valeurs des variables de test.

LinearRegression et train_test_split ont été préchargés à partir de leurs modules respectifs.

Instructions

100 XP
  • Créez X, un tableau contenant les valeurs de toutes les variables explicatives dans sales_df, et y, contenant toutes les valeurs de la colonne "sales".
  • Instanciez un modèle de régression linéaire.
  • Ajustez le modèle aux données d'entraînement.
  • Créez y_pred en faisant des prédictions pour sales à l'aide des variables de test.