Een lineair regressiemodel bouwen
Nu je je feature- en target-arrays hebt gemaakt, ga je een lineair regressiemodel trainen op alle feature- en targetwaarden.
Omdat het doel is om de relatie tussen de feature- en targetwaarden te beoordelen, is het niet nodig de gegevens op te splitsen in trainings- en testsets.
X en y zijn al voor je geladen als volgt:
y = sales_df["sales"].values
X = sales_df["radio"].values.reshape(-1, 1)
Deze oefening maakt deel uit van de cursus
Supervised Learning met scikit-learn
Oefeninstructies
- Importeer
LinearRegression. - Instantieer een lineair regressiemodel.
- Voorspel sales-waarden met
Xen sla deze op alspredictions.
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
# Import LinearRegression
from ____.____ import ____
# Create the model
reg = ____()
# Fit the model to the data
____
# Make predictions
predictions = ____
print(predictions[:5])