Revisão de ensaios

Concluído

Quando utiliza o Optuna para experimentar várias combinações de hiperparâmetros, pode rever os detalhes de cada versão experimental. Pode rever estes detalhes de duas formas:

  • Exiba a execução do MLflow para cada avaliação.
  • Use as classes Estudo e Avaliação para capturar detalhes da execução.

Exibir a execução do MLflow para cada avaliação

No Azure Databricks, você pode integrar o Optuna ao MLflow para gerar automaticamente execuções de experimento MLflow para cada avaliação. Isso fornece uma maneira fácil de exibir o conjunto completo de hiperparâmetros e métricas para cada avaliação no portal do Azure Databricks, conforme mostrado aqui:

Captura de tela da execução do MLflow para uma avaliação do Optuna.

Use as aulas de Estudo e Teste

O Optuna gerencia os detalhes da avaliação usando as classes Estudo e Avaliação .

  • Estudo: Um estudo representa uma sessão de otimização. Ele gerencia o processo geral de encontrar os melhores hiperparâmetros. Você pode pensar nele como um recipiente que armazena todos os ensaios, seus resultados e os melhores parâmetros encontrados até agora.

  • Avaliação: Uma avaliação representa uma única execução do processo de otimização com um conjunto específico de hiperparâmetros. Cada ensaio regista os valores dos parâmetros utilizados, o valor objetivo resultante (como precisão ou perda) e outros metadados (como duração, estado, etc.).

O exemplo de código a seguir mostra como usar o Optuna para revisar os detalhes de cada avaliação.

import optuna

def objective(trial):
    # Define your hyperparameters using trial.suggest_* methods
    x = trial.suggest_float("x", -10, 10)
    # ... your model training and evaluation ...
    return (x - 2) ** 2  # Example objective

study = optuna.create_study(direction="minimize")
study.optimize(objective, n_trials=100)

print("Best param values: ", study.best_params)

# Get details from each trial run
print("trials:")
for trial in study.trials:
    print("\n", trial)

Sugestão

Para obter mais informações sobre os detalhes gravados pelas classes de avaliação , consulte a documentação do Optuna.