ComenzarEmpieza gratis

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

Ver curso

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")
Editar y ejecutar código