ComeçarComece gratuitamente

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

Ver Curso

Instruções de exercício

  • Importe LinearRegression.
  • Instancie um modelo de regressão linear.
  • Preveja os valores de vendas usando X e armazenando como predictions.

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])
Editar e executar código