Module docs.templates.training_model.train

Expand source code
# -------------------------------------------------------------------------------------------------------------
# Script padrão que será chamado dentro da API para realizar o treino do(s) modelo(s) e registro
# no MLflow da API.
#
# Instruções:
# - Organize o seu código dentro da pasta 'training_model', não deixe nada fora desta pasta. Exceto os códigos
#   que devem ficar nas pastas 'worker_pub' e 'worker_retrain';
# - Chame os seus scripts a partir desse script main (train.py).
#
# Dica:
# - Você pode (e deve) criar uma estrutura de pastas dentro da pasta 'training_model' para organizar
#   o seu código, mas faça isso de forma que não haja necessidade de criar a variável de ambiente PYTHONPATH
#   para fazer os imports do código do modelo.
# -------------------------------------------------------------------------------------------------------------

# Faça aqui seus imports...

if __name__ == "__main__":
    # Implemente aqui o seu script de treinamento
    pass