Add GPBandit multiobjective as default by using from_problem factory.

PiperOrigin-RevId: 583067422
google · Nov 16, 2023 · a6bcfc9 · a6bcfc9
1 parent a94eb5a
commit a6bcfc9
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 2 deletions.
diff --git a/vizier/_src/service/policy_factory.py b/vizier/_src/service/policy_factory.py
@@ -35,14 +35,17 @@ def __call__(
  study_name: str,
  ) -> pythia.Policy:
  del study_name
+ # TODO: Compare string to Algorithm enums.
  if algorithm in (
  'DEFAULT',
  'ALGORITHM_UNSPECIFIED',
  'GAUSSIAN_PROCESS_BANDIT',
  ):
  from vizier._src.algorithms.designers import gp_bandit
 
- return dp.DesignerPolicy(policy_supporter, gp_bandit.VizierGPBandit)
+ return dp.DesignerPolicy(
+ policy_supporter, gp_bandit.VizierGPBandit.from_problem
+ )
  elif algorithm == 'RANDOM_SEARCH':
  from vizier._src.algorithms.policies import random_policy
 

diff --git a/vizier/_src/service/vizier_client_test.py b/vizier/_src/service/vizier_client_test.py
@@ -224,12 +224,32 @@ def test_get_suggestions(self):
  dict(algorithm=pyvizier.Algorithm.QUASI_RANDOM_SEARCH),
  dict(algorithm=pyvizier.Algorithm.GRID_SEARCH),
  dict(algorithm=pyvizier.Algorithm.NSGA2, multi_objective=True),
+ dict(
+ algorithm=pyvizier.Algorithm.ALGORITHM_UNSPECIFIED,
+ multi_objective=True,
+ ),
+ dict(
+ algorithm='DEFAULT',
+ multi_objective=True,
+ ),
+ dict(
+ algorithm='DEFAULT',
+ multi_objective=False,
+ ),
+ dict(
+ algorithm=pyvizier.Algorithm.GAUSSIAN_PROCESS_BANDIT,
+ multi_objective=True,
+ ),
+ dict(
+ algorithm=pyvizier.Algorithm.GAUSSIAN_PROCESS_BANDIT,
+ multi_objective=False,
+ ),
  )
  def test_e2e_tuning(
  self,
  *,
  algorithm,
- num_iterations: int = 50,
+ num_iterations: int = 10,
  batch_size: int = 1,
  multi_objective: bool = False,
  ):