From 097992ba08f659cf06f257b2aa1a019f408af7fc Mon Sep 17 00:00:00 2001 From: vizier-team Date: Wed, 18 Dec 2024 10:44:13 -0800 Subject: [PATCH] Minor refactor of `GP_UCB_PE` PiperOrigin-RevId: 707600656 --- vizier/_src/algorithms/designers/gp_ucb_pe.py | 31 ++++++++++--------- 1 file changed, 16 insertions(+), 15 deletions(-) diff --git a/vizier/_src/algorithms/designers/gp_ucb_pe.py b/vizier/_src/algorithms/designers/gp_ucb_pe.py index 068f1204d..6fecf0877 100644 --- a/vizier/_src/algorithms/designers/gp_ucb_pe.py +++ b/vizier/_src/algorithms/designers/gp_ucb_pe.py @@ -510,23 +510,24 @@ class method that takes `ModelInput` and returns a factory=padding.PaddingSchedule, kw_only=True ) + default_eagle_config = es.EagleStrategyConfig( + visibility=3.6782451729470043, + gravity=3.028167342024462, + negative_gravity=0.03036267153343141, + perturbation=0.23337470891647027, + categorical_perturbation_factor=9.587350648631066, + pure_categorical_perturbation_factor=28.636337967676518, + prob_same_category_without_perturbation=0.9744882009359648, + perturbation_lower_bound=7.376256294543107e-4, + penalize_factor=0.7817632796830948, + pool_size_exponent=2.0494446726436744, + mutate_normalization_type=es.MutateNormalizationType.RANDOM, + normalization_scale=1.9893618760239418, + prior_trials_pool_pct=0.423499384081575, + ) default_acquisition_optimizer_factory = vb.VectorizedOptimizerFactory( strategy_factory=es.VectorizedEagleStrategyFactory( - eagle_config=es.EagleStrategyConfig( - visibility=3.6782451729470043, - gravity=3.028167342024462, - negative_gravity=0.03036267153343141, - perturbation=0.23337470891647027, - categorical_perturbation_factor=9.587350648631066, - pure_categorical_perturbation_factor=28.636337967676518, - prob_same_category_without_perturbation=0.9744882009359648, - perturbation_lower_bound=7.376256294543107e-4, - penalize_factor=0.7817632796830948, - pool_size_exponent=2.0494446726436744, - mutate_normalization_type=es.MutateNormalizationType.RANDOM, - normalization_scale=1.9893618760239418, - prior_trials_pool_pct=0.423499384081575, - ) + eagle_config=default_eagle_config ), max_evaluations=75000, suggestion_batch_size=25,