Updates the default value for pe_overwrite_probability

PiperOrigin-RevId: 627118111
google · Apr 22, 2024 · 00ef185 · 00ef185
1 parent 41f2bcc
commit 00ef185
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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):