From 2067c659b477a219c4a9c0dbb6f0f5af710af539 Mon Sep 17 00:00:00 2001 From: victor Date: Wed, 29 Sep 2021 11:44:53 +0100 Subject: [PATCH] fixing reference point etc. --- trieste/acquisition/function.py | 2 +- trieste/bayesian_optimizer.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/trieste/acquisition/function.py b/trieste/acquisition/function.py index 19d72cc1aa..7e7cd0eccb 100644 --- a/trieste/acquisition/function.py +++ b/trieste/acquisition/function.py @@ -1060,7 +1060,7 @@ def prepare_acquisition_function( feasible_mean, _ = objective_model.predict(feasible_query_points) _pf = Pareto(feasible_mean) - _reference_pt = get_reference_point(_pf.front) + _reference_pt = get_reference_point(feasible_mean) ehvi = expected_hv_improvement(objective_model, _pf, _reference_pt) return lambda at: ehvi(at) * constraint_fn(at) diff --git a/trieste/bayesian_optimizer.py b/trieste/bayesian_optimizer.py index 4de61ed3a5..0089ee6b3a 100644 --- a/trieste/bayesian_optimizer.py +++ b/trieste/bayesian_optimizer.py @@ -352,6 +352,7 @@ def optimize( history: list[Record[S]] = [] for step in range(num_steps): + print(step) try: if track_state: models_copy = copy.deepcopy(models)