Registrar una ejecución
En este ejercicio, entrenarás un modelo utilizando Linear Regression de scikit-learn para predecir el beneficio a partir del conjunto de datos Unicorn. Has creado un experimento llamado Unicorn Sklearn Experiment y has iniciado una nueva ejecución. Registrarás métricas para r2_score y parámetros para n_jobs, además de registrar el código de entrenamiento como artefacto.
El modelo Linear Regression se entrenará con el parámetro n_jobs ajustado a 1. La métrica r2_score se ha elaborado utilizando la r2_score() de scikit-learn basada en la variable y_pred que procede de las predicciones de X_test.
model = LinearRegression(n_jobs=1)
model.fit(X_train, y_train)
y_pred = model.predict(X_test)
r2_score = r2_score(y_test, y_pred)
Se importarán el módulo mlflow y los módulos LinearRegression, train_test_split y metrics de scikit-learn.
Este ejercicio forma parte del curso
Introducción a MLflow
Instrucciones del ejercicio
- Registra la variable
r2_scorecomo una métrica denominada"r2_score". - Registra un parámetro llamado
"n_jobs"en el Servidor de Seguimiento. - Registra el archivo
"train.py"como artefacto en la ejecución.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# Log the metric r2_score as "r2_score"
____.____("____", ____)
# Log parameter n_jobs as "n_jobs"
____.____("____", ____)
# Log the training code
____.____("train.py")