Registro de modelos ajustados
Has estado probando distintos hiperparámetros del modelo y necesitas registrar tu última ronda de resultados del experimento en MLflow. ¡Hagámoslo!
Este ejercicio forma parte del curso
Diseño de canalizaciones de predicción para producción
Instrucciones del ejercicio
- Establece el nombre del experimento como
"hyperparameter_tuning". - Itera sobre el índice y las filas de
df. - Inicia una ejecución de MLflow.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# Set the experiment name
experiment_name = "____"
experiment_id = mlflow.create_experiment(experiment_name)
# Loop through the DataFrame
for idx, row in df.____():
# Start a run
with mlflow.____(experiment_id=____):
model_params = ml_models[row["model_label"]].get_params()
model_params["model_name"] = row["model_name"]
model_params["model_label"] = row["model_label"]
model_params["partition"] = row["partition"]
model_params["lags"] = list(range(1, 24))
model_params["date_features"] = ["month", "day", "dayofweek", "week", "hour"]
mlflow.log_params(model_params)
mlflow.log_metric("mape", row["mape"])
mlflow.log_metric("rmse", row["rmse"])