Construire un modèle de régression linéaire
Maintenant que vous avez créé vos tableaux de caractéristiques et de cibles, vous allez former un modèle de régression linéaire sur toutes les valeurs des caractéristiques et des cibles.
L’objectif étant d’évaluer la relation entre les caractéristiques et les valeurs cibles, il n’est pas nécessaire de diviser les données en ensembles d’apprentissage et de test.
X
et y
ont été préchargés pour vous comme suit :
y = sales_df["sales"].values
X = sales_df["radio"].values.reshape(-1, 1)
Cet exercice fait partie du cours
Apprentissage supervisé avec scikit-learn
Instructions
- Importez
LinearRegression
. - Instanciez un modèle de régression linéaire.
- Faites une prédiction des valeurs de ventes à l’aide de
X
, en les enregistrant souspredictions
.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# Import LinearRegression
from ____.____ import ____
# Create the model
reg = ____()
# Fit the model to the data
____
# Make predictions
predictions = ____
print(predictions[:5])