Realiza un ejercicio práctico sencillo para aplicar técnicas de aprendizaje supervisado utilizando un modelo de regresión lineal. Utiliza un pequeño conjunto de datos ficticio que crearás en Python para entrenar y evaluar un modelo de regresión lineal utilizando scikit-learn
.
Utiliza Python para crear un pequeño conjunto de datos con dos características y un objetivo.
Este código proporciona una base sólida para comenzar a trabajar con regresión lineal en Python.
🌟 Prueba este código en Google Colab
import pandas as pd
from sklearn.linear_model import LinearRegression
# 1. Crear el conjunto de datos
data = {
'horas_estudio': [2, 3, 5, 7, 8],
'examen_practico': [50, 60, 75, 80, 85],
'nota_final': [58, 65, 72, 78, 83]
}
df = pd.DataFrame(data)
# 2. Entrenar un modelo de regresión lineal
X = df[['horas_estudio', 'examen_practico']] # Características (variables independientes)
y = df['nota_final'] # Variable objetivo (a predecir)
model = LinearRegression()
model.fit(X, y)
# 3. Predecir con nuevos datos
nuevos_datos = [[6, 70]] # Un nuevo estudiante con 6 horas de estudio y 70 en el examen práctico
prediccion = model.predict(nuevos_datos)
print("Predicción de la Nota Final:", prediccion)
pandas
para manejar los datos y sklearn.linear_model
para utilizar el modelo de regresión lineal.X
: Contiene las características que se utilizarán para hacer las predicciones.y
: Contiene la variable objetivo que queremos predecir.fit
.nuevos_datos
.matplotlib
o seaborn
para crear gráficos.