ComeçarComece gratuitamente

Registro de uma execução

Neste exercício, você treinará um modelo usando o Linear Regression do scikit-learn para prever o lucro do conjunto de dados Unicorn. Você criou um experimento chamado Unicorn Sklearn Experiment e iniciou uma nova execução. Você registrará métricas para r2_score e parâmetros para n_jobs, além de registrar o código de treinamento como um artefato.

O modelo Linear Regression será treinado com o parâmetro n_jobs definido como 1. A métrica r2_score foi produzida usando o r2_score() do scikit-learn com base na variável y_pred que veio das previsões do 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)

O módulo mlflow, bem como os módulos LinearRegression, train_test_split e metrics do scikit-learn serão importados.

Este exercício faz parte do curso

Introdução ao fluxo ML

Ver Curso

Instruções de exercício

  • Registre a variável r2_score como uma métrica chamada "r2_score".
  • Registre um parâmetro chamado "n_jobs" no Servidor de Rastreamento.
  • Registre o arquivo "train.py" como um artefato na execução.

Exercício interativo prático

Experimente este exercício preenchendo este código de exemplo.

# Log the metric r2_score as "r2_score"
____.____("____", ____)

# Log parameter n_jobs as "n_jobs"
____.____("____", ____)

# Log the training code
____.____("train.py")
Editar e executar código