swim/bo-ptycho
1
0
Fork 0
mirror of https://github.com/c-sooyoung/bo-ptycho.git synced 2026-09-17 21:29:07 +09:00
Code Issues Packages Projects Releases Wiki Activity

added hotfix for batched bo

Browse Source
This commit is contained in:
swim
2026-08-12 14:50:05 +09:00
parent c638cc57a0
commit 781ec07b1e
7 changed files with 102 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files
bo/random.py
+14 -11
View File
@@ -37,20 +37,23 @@ class RandomBOEngine(BOEngine):
self.train_y = train_y
def ask(self):
def ask(self, n = 1):
config = self.config
next_config = copy.deepcopy(config)
next_configs = []
for param in self.params:
radius = config['bo']['params'][param]['radius']
center = config['ptycho']['params'][param]
modulation = radius * (np.random.rand() - 0.5) * 2
next_value = center + modulation
if self.param_types[param] == 'int':
next_value = round(next_value)
next_config['ptycho']['params'][param] = next_value
for _ in range(n):
next_config = copy.deepcopy(config)
for param in self.params:
radius = config['bo']['params'][param]['radius']
center = config['ptycho']['params'][param]
modulation = radius * (np.random.rand() - 0.5) * 2
next_value = center + modulation
if self.param_types[param] == 'int':
next_value = round(next_value)
next_config['ptycho']['params'][param] = next_value
next_configs.append(next_config)
return next_config
return next_configs
def tell(self, job_config, y_value):