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
Instruções do exercício
- Registre a variável
r2_scorecomo 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 completando 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")