CommencerCommencer gratuitement

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

Afficher le cours

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 sous predictions.

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])
Modifier et exécuter le code