Increases the default signal-to-noise threshold in GP-UCB-PE.

vizier-team · copybara-github · commit dbbaaf94dcc9 · 2025-07-11T09:24:19.000-07:00
PiperOrigin-RevId: 782007942
diff --git a/vizier/_src/algorithms/designers/gp_ucb_pe.py b/vizier/_src/algorithms/designers/gp_ucb_pe.py
@@ -106,7 +106,7 @@ class UCBPEConfig(eqx.Module):
   # is below this threshold, the designer considers the noise to be high and may
   # explore more aggressively. Set to 0.0 to disable this feature.
   signal_to_noise_threshold: jt.Float[jt.Array, ''] = eqx.field(
-      default=0.1, converter=jnp.asarray
+      default=0.7, converter=jnp.asarray
   )
   # Whether to optimize the set acquisition function for exploration.
   optimize_set_acquisition_for_exploration: bool = eqx.field(

Original file line number	Diff line number	Diff line change
`@@ -106,7 +106,7 @@ class UCBPEConfig(eqx.Module):`
`106`	`106`	`# is below this threshold, the designer considers the noise to be high and may`
`107`	`107`	`# explore more aggressively. Set to 0.0 to disable this feature.`
`108`	`108`	`signal_to_noise_threshold: jt.Float[jt.Array, ''] = eqx.field(`
`109`		`- default=0.1, converter=jnp.asarray`
	`109`	`+ default=0.7, converter=jnp.asarray`
`110`	`110`	`)`
`111`	`111`	`# Whether to optimize the set acquisition function for exploration.`
`112`	`112`	`optimize_set_acquisition_for_exploration: bool = eqx.field(`