Adicionando parâmetros ao MLproject
Definir parâmetros em MLflow Projects permite tornar seu código de ML reproduzível. Parâmetros também simplificam a execução de experimentos de treinamento com configurações diferentes sem precisar alterar o código.
Neste exercício, você vai adicionar parâmetros ao seu arquivo MLproject para o ponto de entrada principal. Esse ponto de entrada é usado para executar o script train_model.py, que treina um modelo de Logistic Regression a partir de dados de Insurance.
O script aceita dois parâmetros, n_jobs e fit_intercept, que são hiperparâmetros usados para treinar o modelo. Você começará adicionando o parâmetro n_jobs no arquivo MLproject. Depois, adicionará o parâmetro fit_intercept. Por fim, você incluirá os parâmetros no comando executado no ponto de entrada principal.
Este exercício faz parte do curso
Introdução ao MLflow
Instruções do exercício
- Crie um parâmetro chamado
n_jobsdo tipointe valor padrão1. - Crie um segundo parâmetro chamado
fit_interceptdo tipoboolcom valor padrão definido comoTrue. - Passe ambos os parâmetros para o comando, garantindo que
n_jobsvenha primeiro seguido defit_intercept.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
"""
name: insurance_model
python_env: python_env.yaml
entry_points:
main:
parameters:
# Create parameter for number of jobs as n_jobs
____:
____: ____
____: ____
# Create parameter for fit_intercept
____:
____: ____
____: ____
# Add parameters to be passed into the command
command: "python3.9 train_model.py {____} {____}"
"""