def optuna_objective_function(trial):
hidden_layer_size=trial.suggest_int("hidden_layer_size", 32, 512, 1),
activation_function=trial.suggest_categorical("activation_function", ["tanh", "relu"]),
optuna_run = Run(number_of_epochs=NUMBER_OF_EPOCHS, run_type="optuna search")
metric_value = optuna_run.execute(args)
study = optuna.create_study(direction="maximize")
study.optimize(optuna_objective_function, n_trials=BUDGET, show_progress_bar=False)