From 00ef185df47d439c6681c2e94cfe649a52486d4c Mon Sep 17 00:00:00 2001 From: vizier-team Date: Mon, 22 Apr 2024 12:17:40 -0700 Subject: [PATCH] Updates the default value for `pe_overwrite_probability` PiperOrigin-RevId: 627118111 --- vizier/_src/algorithms/designers/gp_ucb_pe.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/vizier/_src/algorithms/designers/gp_ucb_pe.py b/vizier/_src/algorithms/designers/gp_ucb_pe.py index 9d80ae0fa..2145e5dd9 100644 --- a/vizier/_src/algorithms/designers/gp_ucb_pe.py +++ b/vizier/_src/algorithms/designers/gp_ucb_pe.py @@ -71,11 +71,10 @@ class UCBPEConfig(eqx.Module): ucb_overwrite_probability: jt.Float[jt.Array, ''] = eqx.field( default=0.25, converter=jnp.asarray ) - # Probability of selecting the PE acquisition function when there are new # completed trials. pe_overwrite_probability: jt.Float[jt.Array, ''] = eqx.field( - default=0.0, converter=jnp.asarray + default=0.1, converter=jnp.asarray ) def __repr__(self):