Construir un modelo de regresión lineal
Ahora que has creado tus matrices de características y objetivos, entrenarás un modelo de regresión lineal en todos los valores de características y objetivos.
Como el objetivo es evaluar la relación entre la característica y los valores objetivo, no es necesario dividir los datos en conjuntos de entrenamiento y de prueba.
X
y y
se han precargado para ti de la siguiente manera:
y = sales_df["sales"].values
X = sales_df["radio"].values.reshape(-1, 1)
Este ejercicio forma parte del curso
Aprendizaje supervisado con scikit-learn
Instrucciones de ejercicio
- Importa
LinearRegression
. - Instanciar un modelo de regresión lineal.
- Predecir los valores de venta utilizando
X
, almacenando comopredictions
.
Ejercicio interactivo práctico
Pruebe este ejercicio completando este código de muestra.
# Import LinearRegression
from ____.____ import ____
# Create the model
reg = ____()
# Fit the model to the data
____
# Make predictions
predictions = ____
print(predictions[:5])