From 4be7df11eca1f5726bfcc1760beda182932b1680 Mon Sep 17 00:00:00 2001 From: Kevin Stone Date: Wed, 21 Aug 2024 10:51:58 -0400 Subject: [PATCH] Added default weights to NParEGO --- obsidian/optimizer/bayesian.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/obsidian/optimizer/bayesian.py b/obsidian/optimizer/bayesian.py index 6e16cdf..df7c122 100644 --- a/obsidian/optimizer/bayesian.py +++ b/obsidian/optimizer/bayesian.py @@ -459,8 +459,8 @@ def _validate_hypers(self, if hps.get(key) is None: if not defaults['optional']: raise ValueError(f'Must specify hyperpameter value {key} for {aq_str}') - if key in ['weights']: #['scalarization_weights', 'weights']: - aq_hps[key] = defaults['val'] * o_dim + if key in ['weights', 'scalarization_weights']: + aq_hps[key] = [1] * o_dim else: aq_hps[key] = defaults['val'] @@ -553,7 +553,7 @@ def _parse_aq_kwargs(self, if aq == 'NParEGO': w = hps['scalarization_weights'] - if isinstance(w,list): + if isinstance(w, list): w = torch.tensor(w) w = w/torch.sum(torch.abs(w)) aq_kwargs['scalarization_weights'] = w