Bayesian exploration from YAML¶

In [1]:

from xopt import Xopt

# set values if testing
import os

import warnings

warnings.filterwarnings("ignore")


SMOKE_TEST = os.environ.get("SMOKE_TEST")

YAML = """
generator:
    name: bayesian_exploration

evaluator:
    function: xopt.resources.test_functions.tnk.evaluate_TNK

vocs:
    variables:
        x1: [0, 3.14159]
        x2: [0, 3.14159]
    observables: [y1]
    constraints:
        c1: [GREATER_THAN, 0]
        c2: [LESS_THAN, 0.5]
    constants: {a: dummy_constant}

"""

from xopt import Xopt # set values if testing import os import warnings warnings.filterwarnings("ignore") SMOKE_TEST = os.environ.get("SMOKE_TEST") YAML = """ generator: name: bayesian_exploration evaluator: function: xopt.resources.test_functions.tnk.evaluate_TNK vocs: variables: x1: [0, 3.14159] x2: [0, 3.14159] observables: [y1] constraints: c1: [GREATER_THAN, 0] c2: [LESS_THAN, 0.5] constants: {a: dummy_constant} """

In [2]:

X = Xopt.from_yaml(YAML)

# for testing purposes only
if SMOKE_TEST:
    X.generator.numerical_optimizer.n_restarts = 1
    X.generator.n_monte_carlo_samples = 1

X

X = Xopt.from_yaml(YAML) # for testing purposes only if SMOKE_TEST: X.generator.numerical_optimizer.n_restarts = 1 X.generator.n_monte_carlo_samples = 1 X

Out[2]:

            Xopt
________________________________
Version: 2.6.8.dev18+g6fb143c55.d20251203
Data size: 0
Config as YAML:
dump_file: null
evaluator:
  function: xopt.resources.test_functions.tnk.evaluate_TNK
  function_kwargs:
    raise_probability: 0
    random_sleep: 0
    sleep: 0
  max_workers: 1
  vectorized: false
generator:
  computation_time: null
  custom_objective: null
  fixed_features: null
  gp_constructor:
    covar_modules: {}
    custom_noise_prior: null
    mean_modules: {}
    name: standard
    trainable_mean_keys: []
    transform_inputs: true
    use_cached_hyperparameters: false
    use_low_noise_prior: false
  max_travel_distances: null
  model: null
  n_candidates: 1
  n_interpolate_points: null
  n_monte_carlo_samples: 128
  name: bayesian_exploration
  numerical_optimizer:
    max_iter: 2000
    max_time: 5.0
    n_restarts: 20
    name: LBFGS
  supports_batch_generation: true
  supports_constraints: true
  turbo_controller: null
  use_cuda: false
max_evaluations: null
serialize_inline: false
serialize_torch: false
strict: true
vocs:
  constants:
    a: dummy_constant
  constraints:
    c1:
    - GREATER_THAN
    - 0.0
    c2:
    - LESS_THAN
    - 0.5
  objectives: {}
  observables:
  - y1
  variables:
    x1:
    - 0.0
    - 3.14159
    x2:
    - 0.0
    - 3.14159

In [3]:

X.random_evaluate(5)

for i in range(5):
    print(f"step {i}")
    X.step()

X.random_evaluate(5) for i in range(5): print(f"step {i}") X.step()

step 0

step 1

step 2

step 3

step 4

In [4]:

print(X.data)

print(X.data)

         x1        x2               a        y1        y2         c1  \
0  1.335564  2.139286  dummy_constant  1.335564  2.139286   5.448264   
1  0.944680  3.136328  dummy_constant  0.944680  3.136328   9.732114   
2  1.947369  0.625683  dummy_constant  1.947369  0.625683   3.157855   
3  1.168174  2.676671  dummy_constant  1.168174  2.676671   7.433695   
4  1.908079  1.237096  dummy_constant  1.908079  1.237096   4.268730   
5  3.141590  3.141590  dummy_constant  3.141590  3.141590  18.639175   
6  0.000000  0.000000  dummy_constant  0.000000  0.000000  -1.100000   
7  1.034821  1.099738  dummy_constant  1.034821  1.099738   1.191877   
8  0.822871  0.850095  dummy_constant  0.822871  0.850095   0.303148   
9  1.329791  0.509297  dummy_constant  1.329791  0.509297   0.936876   

          c2  xopt_runtime  xopt_error  
0   3.385426      0.000226       False  
1   7.147967      0.000138       False  
2   2.110672      0.000137       False  
3   5.184354      0.000129       False  
4   2.525996      0.000129       False  
5  13.955995      0.000153       False  
6   0.500000      0.000150       False  
7   0.645718      0.000153       False  
8   0.226812      0.000153       False  
9   0.688640      0.000151       False  

In [5]:

# plot results
ax = X.data.plot("x1", "x2")
ax.set_aspect("equal")

# plot results ax = X.data.plot("x1", "x2") ax.set_aspect("equal")

In [6]:

fig, ax = X.generator.visualize_model(show_feasibility=True, n_grid=100)

fig, ax = X.generator.visualize_model(show_feasibility=True, n_grid=100)