loskuri / Meli_Real_Estate_Predict

Prediccion de precio de propiedades en Amba con la base de mercado librew
0 stars 0 forks source link

Correr modelos #4

Open loskuri opened 1 month ago

loskuri commented 1 month ago

CREAR FUNCIONES POR EJEMPLO:

def modelo_lasso(df, vars, params):
   # Corre el lasso, no elige hyperparametros. SIrve para entrenar un modelo solo.
   metrics = calcula_metricas(df[y], model)
   # Guardar params y metrics

   return model, metrics

def correr_lasso(df, vars): 
   grilla = ...
   model = modelo_lasso(df, vars, params=grilla):
   return metrics, model

Metrics son las métricas que vamos a usar (MSE) y model es el modelo entrenado para poder reutilizarlo.