weight-predictor

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]])