weight-predictor / README.md
elsamueldev's picture
Update README.md
ba9bcd0

Modelo de Regresi贸n Lineal para Predicci贸n de Peso

En 茅ste repo se encuentra el modelo que he creado al completar una de las pr谩cticas del curso "Introducci贸n a la IA con Python", de mano de Rodrigo Montemayor, o "Ringa Tech", como bien se conoce en redes sociales.

C贸mo se usa

El modelo ha sido creado y entrenado con scikit-learn y exportado con joblib, por lo que necesitas ambas librer铆as instaladas. Vienen por defecto en los entornos de Google Colab, si gustas probarlo all铆.

Cuando tengas las librer铆as y el archivo .joblib del modelo descargado, usa la funci贸n load() pas谩ndole la ruta al archivo

from joblib import load

model = load("modelo.joblib")

Una vez cargado, puedes hacer predicciones usando su m茅todo .predict(), pero recuerda que s贸lo admite arrays de 2 dimensiones (uno dentro de otro, vaya). El dato de entrada es la altura en metros, y el de salida es el peso en kilos

weight = model.predict([[1.65]])