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.
Cette activité fait partie du cours
Apprentissage supervisé avec scikit-learn
Instructions de l’exercice
- Créez
X, un tableau contenant les valeurs de toutes les variables explicatives danssales_df, ety, 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_preden faisant des prédictions poursalesà l'aide des variables de test.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# Create X and y arrays
X = sales_df.____("____", axis=____).____
y = sales_df["____"].____
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)
# Instantiate the model
reg = ____
# Fit the model to the data
____
# Make predictions
y_pred = reg.____(____)
print("Predictions: {}, Actual Values: {}".format(y_pred[:2], y_test[:2]))