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 de exercício
- Crie um parâmetro chamado
n_jobs
como um tipoint
e um valor padrão de1
. - Crie um segundo parâmetro chamado
fit_intercept
como um tipobool
com um valor padrão definido comoTrue
. - Passe os dois parâmetros para o comando, garantindo que
n_jobs
seja o primeiro, seguido porfit_intercept
.
Exercício interativo prático
Experimente este exercício preenchendo 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 {____} {____}"
"""