Registrar dados automaticamente em uma execução de experimento

O registro automático é um recurso do SDK da Vertex AI que registra automaticamente os parâmetros e métricas das execuções de treinamento de modelo para os experimentos da Vertex AI. Isso pode economizar tempo e esforço, eliminando a necessidade de registrar esses dados manualmente. Atualmente, o registro automático é compatível apenas com a geração de registros de parâmetros e métricas.

Registrar dados automaticamente

Há duas opções para registrar dados automaticamente nos experimentos da Vertex AI.

1. Deixe o SDK Vertex AI criar recursos do ExperimentRun automaticamente para você.
   
2. Especifique o recurso ExperimentRun em que você gostaria que as métricas e os parâmetros fossem registrados automaticamente.
   

def autologging_with_auto_run_creation_sample(
    experiment_name: str,
    project: str,
    location: str,
    experiment_tensorboard: Optional[Union[str, aiplatform.Tensorboard]] = None,
):
    aiplatform.init(
        experiment=experiment_name,
        project=project,
        location=location,
        experiment_tensorboard=experiment_tensorboard,
    )

    aiplatform.autolog()

    # Your model training code goes here

    aiplatform.autolog(disable=True)

def autologging_with_manual_run_creation_sample(
    experiment_name: str,
    run_name: str,
    project: str,
    location: str,
    experiment_tensorboard: Optional[Union[str, aiplatform.Tensorboard]] = None,
):
    aiplatform.init(
        experiment=experiment_name,
        project=project,
        location=location,
        experiment_tensorboard=experiment_tensorboard,
    )

    aiplatform.autolog()

    aiplatform.start_run(run=run_name)

    # Your model training code goes here

    aiplatform.end_run()

    aiplatform.autolog(disable=True)

O registro automático do SDK da Vertex AI usa o registro automático do MLFlow na implementação. As métricas e os parâmetros de avaliação dos frameworks a seguir são registrados no ExperimentRun quando a geração automática de registros está ativada.

• Fastai
• Gluon
• Keras
• LightGBM
• Pytorch Lightning
• Scikit-learn
• Spark
• Statsmodels
• XGBoost

Ver métricas e parâmetros registrados automaticamente

Use o SDK da Vertex AI para Python para comparar execuções e receber dados de execuções. O Console do Google Cloud oferece uma maneira fácil de comparar essas execuções.

Exemplo de notebook relevante

• Registrar dados automaticamente

Postagem do blog

• Como automatizar o acompanhamento de experimentos de ML com a geração automática de registros dos experimentos da Vertex AI