Adicionando parâmetros ao projeto ML
A definição de parâmetros nos projetos de fluxo do MLpermite que você torne seu código ML reproduzível. Os parâmetros também simplificam a execução de experimentos de treinamento com configurações diferentes sem a necessidade de alterar o código.
Neste exercício, você 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 regressão logística a partir dos dados do seguro.
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. Em seguida, você adicionará o parâmetro fit_intercept. Por fim, você adicionará os parâmetros ao comando executado no ponto de entrada principal.
Este exercício faz parte do curso
Introdução ao fluxo ML
Instruções do exercício
- Crie um parâmetro chamado
n_jobscomo um tipointe um valor padrão de1. - Crie um segundo parâmetro chamado
fit_interceptcomo um tipoboolcom um valor padrão definido comoTrue. - Passe os dois parâmetros para o comando, garantindo que
n_jobsseja o primeiro, seguido porfit_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 {____} {____}"
"""