Add GP_UCB_PE and make it the new default.

PiperOrigin-RevId: 601196580

README.md

@@ -36,7 +36,7 @@ def evaluate(w: float, x: int, y: float, z: str) -> float:
  return w**2 - y**2 + x * ord(z)
 
 # Algorithm, search space, and metrics.
-study_config = vz.StudyConfig(algorithm='GAUSSIAN_PROCESS_BANDIT')
+study_config = vz.StudyConfig(algorithm='DEFAULT')
 study_config.search_space.root.add_float_param('w', 0.0, 5.0)
 study_config.search_space.root.add_int_param('x', -2, 2)
 study_config.search_space.root.add_discrete_param('y', [0.3, 7.2])
@@ -46,7 +46,7 @@ study_config.metric_information.append(vz.MetricInformation('metric_name', goal=
 # Setup client and begin optimization. Vizier Service will be implicitly created.
 study = clients.Study.from_study_config(study_config, owner='my_name', study_id='example')
 for i in range(10):
- suggestions = study.suggest(count=1)
+ suggestions = study.suggest(count=2)
  for suggestion in suggestions:
  params = suggestion.parameters
  objective = evaluate(params['w'], params['x'], params['y'], params['z'])

docs/guides/user/running_vizier.ipynb

@@ -108,7 +108,7 @@
  "outputs": [],
  "source": [
  "study_config = vz.StudyConfig.from_problem(problem)\n",
- "study_config.algorithm = 'GAUSSIAN_PROCESS_BANDIT'"
+ "study_config.algorithm = 'DEFAULT'"
  ]
  },
  {

docs/guides/user/supported_algorithms.ipynb

@@ -19,12 +19,13 @@
  "## Official\n",
  "The following algorithms can be considered \"official\" and production-quality:\n",
  "\n",
- "1. [**GP-Bandit**](https://github.com/google/vizier/blob/main/vizier/_src/algorithms/designers/gp_bandit.py) (`GAUSSIAN_PROCESS_BANDIT`): Flat Search Spaces.\n",
- "2. [**Random Search**](https://github.com/google/vizier/blob/main/vizier/_src/algorithms/designers/random.py) (`RANDOM_SEARCH`): Flat Search Spaces.\n",
- "3. [**Quasi-Random Search**](https://github.com/google/vizier/blob/main/vizier/_src/algorithms/designers/quasi_random.py) (`QUASI_RANDOM_SEARCH`): Flat Search Spaces.\n",
- "4. [**Grid Search**](https://github.com/google/vizier/blob/main/vizier/_src/algorithms/designers/grid.py) (`GRID_SEARCH`): Flat Search Spaces.\n",
- "5. [**Shuffled Grid Search**](https://github.com/google/vizier/blob/main/vizier/_src/algorithms/designers/grid.py) (`SHUFFLED_GRID_SEARCH`): Flat Search Spaces.\n",
- "6. [**Eagle Strategy**](https://github.com/google/vizier/blob/main/vizier/_src/algorithms/designers/eagle_strategy/eagle_strategy.py) (`EAGLE_STRATEGY`): Flat Search Spaces."
+ "1. [**GP-UCB-PE**](https://github.com/google/vizier/blob/main/vizier/_src/algorithms/designers/gp_ucb_pe.py) (`GP_UCB_PE`): Flat Search Spaces.\n",
+ "2. [**GP-Bandit**](https://github.com/google/vizier/blob/main/vizier/_src/algorithms/designers/gp_bandit.py) (`GAUSSIAN_PROCESS_BANDIT`): Flat Search Spaces.\n",
+ "3. [**Random Search**](https://github.com/google/vizier/blob/main/vizier/_src/algorithms/designers/random.py) (`RANDOM_SEARCH`): Flat Search Spaces.\n",
+ "4. [**Quasi-Random Search**](https://github.com/google/vizier/blob/main/vizier/_src/algorithms/designers/quasi_random.py) (`QUASI_RANDOM_SEARCH`): Flat Search Spaces.\n",
+ "5. [**Grid Search**](https://github.com/google/vizier/blob/main/vizier/_src/algorithms/designers/grid.py) (`GRID_SEARCH`): Flat Search Spaces.\n",
+ "6. [**Shuffled Grid Search**](https://github.com/google/vizier/blob/main/vizier/_src/algorithms/designers/grid.py) (`SHUFFLED_GRID_SEARCH`): Flat Search Spaces.\n",
+ "7. [**Eagle Strategy**](https://github.com/google/vizier/blob/main/vizier/_src/algorithms/designers/eagle_strategy/eagle_strategy.py) (`EAGLE_STRATEGY`): Flat Search Spaces."
  ]
  },
  {

vizier/__init__.py

@@ -23,4 +23,4 @@
 
 sys.path.append(PROTO_ROOT)
 
-__version__ = "0.1.14"
+__version__ = "0.1.15"