ComeçarComece de graça

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

Ver curso

Instruções do exercício

  • Crie um parâmetro chamado n_jobs do tipo int e valor padrão 1.
  • Crie um segundo parâmetro chamado fit_intercept do tipo bool com valor padrão definido como True.
  • Passe ambos os parâmetros para o comando, garantindo que n_jobs venha primeiro seguido de fit_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 {____} {____}"
"""
Editar e executar o código