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 de ejercicio
- Registra la variable
r2_score
como 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
Pruebe este ejercicio completando este 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")