Building a linear regression model
Now you have created your feature and target arrays, you will train a linear regression model on all feature and target values.
As the goal is to assess the relationship between the feature and target values there is no need to split the data into training and test sets.
X
and y
have been preloaded for you as follows:
y = sales_df["sales"].values
X = sales_df["radio"].values.reshape(-1, 1)
This exercise is part of the course
Supervised Learning with scikit-learn
Exercise instructions
- Import
LinearRegression
. - Instantiate a linear regression model.
- Predict sales values using
X
, storing aspredictions
.
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
# Import LinearRegression
from ____.____ import ____
# Create the model
reg = ____()
# Fit the model to the data
____
# Make predictions
predictions = ____
print(predictions[:5])