Criação de um modelo de regressão linear
Agora que você criou suas matrizes de variáveis independentes e dependentes, deve treinar um modelo de regressão linear com base em todos os valores das variáveis independentes e dependentes.
Como o objetivo é avaliar a relação entre os valores das variáveis independentes e dependentes, não há necessidade de dividir os dados em conjuntos de treinamento e teste.
X
e y
foram pré-carregados para você da seguinte forma:
y = sales_df["sales"].values
X = sales_df["radio"].values.reshape(-1, 1)
Este exercício faz parte do curso
Aprendizado Supervisionado com o scikit-learn
Instruções de exercício
- Importe
LinearRegression
. - Instancie um modelo de regressão linear.
- Preveja os valores de vendas usando
X
e armazenando comopredictions
.
Exercício interativo prático
Experimente este exercício preenchendo este código de exemplo.
# Import LinearRegression
from ____.____ import ____
# Create the model
reg = ____()
# Fit the model to the data
____
# Make predictions
predictions = ____
print(predictions[:5])