weight-predictor / README.md
elsamueldev's picture
Update README.md
ba9bcd0
|
raw
history blame
1.11 kB
# 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](https://www.domestika.org/es/courses/5239-introduccion-a-la-ia-con-python/ringatech)", de mano de Rodrigo Montemayor, o "[Ringa Tech](https://youtube.com/@RingaTech)", 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
```python
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
```python
weight = model.predict([[1.65]])
```