Kom igångKom igång gratis

Bygga en linjär regressionsmodell

Nu när du har skapat dina särdragsarrayer och målarray tränar du en linjär regressionsmodell på alla särdragets- och målvärden.

Eftersom målet är att undersöka sambandet mellan särdrag och målvärden behöver du inte dela upp datan i tränings- och testmängder.

X och y har förhandsladdats åt dig enligt följande:

y = sales_df["sales"].values
X = sales_df["radio"].values.reshape(-1, 1)

Den här övningen är en del av kursen

Övervakad inlärning med scikit-learn

Visa kurs

Övningsinstruktioner

  • Importera LinearRegression.
  • Skapa en instans av en linjär regressionsmodell.
  • Förutsäg försäljningsvärden med X och lagra resultatet som predictions.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

# Import LinearRegression
from ____.____ import ____

# Create the model
reg = ____()

# Fit the model to the data
____

# Make predictions
predictions = ____

print(predictions[:5])
Redigera och kör kod