Bayesian exploration from YAML¶
In [1]:
Copied!
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]:
Copied!
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: 0.1.dev1947+g7831d28.d20250426 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]:
Copied!
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]:
Copied!
print(X.data)
print(X.data)
x1 x2 a y1 y2 c1 c2 \ 0 2.074052 0.063556 dummy_constant 2.074052 0.063556 3.217506 2.668124 1 3.061008 0.662200 dummy_constant 3.061008 0.662200 8.904731 6.585072 2 2.359404 1.197353 dummy_constant 2.359404 1.197353 5.967079 3.943685 3 1.364894 0.603272 dummy_constant 1.364894 0.603272 1.133846 0.758708 4 1.631357 2.503001 dummy_constant 1.631357 2.503001 8.024668 5.291980 5 0.000000 1.231559 dummy_constant 0.000000 1.231559 0.416738 0.785179 6 0.582536 1.220076 dummy_constant 0.582536 1.220076 0.761488 0.525321 7 0.813243 0.951859 dummy_constant 0.813243 0.951859 0.536240 0.302297 8 0.447261 1.016210 dummy_constant 0.447261 1.016210 0.138809 0.269255 9 1.026544 0.546468 dummy_constant 1.026544 0.546468 0.349713 0.279408 xopt_runtime xopt_error 0 0.000155 False 1 0.000138 False 2 0.000129 False 3 0.000128 False 4 0.000127 False 5 0.000150 False 6 0.000149 False 7 0.000149 False 8 0.000149 False 9 0.000148 False
In [5]:
Copied!
# 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]:
Copied!
fig, ax = X.generator.visualize_model(show_feasibility=True, n_grid=100)
fig, ax = X.generator.visualize_model(show_feasibility=True, n_grid=100)