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"