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 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")