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

made sobo and randombo more consistent

Browse Source
This commit is contained in:
swim
2026-07-30 15:05:59 +09:00
parent ba4ebfd999
commit 02293389a1
5 changed files with 36 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main.py
+7 -10
View File
@@ -16,30 +16,27 @@ def main(config_yaml):
os.makedirs(result_dir, exist_ok=True)
shutil.copy(config_yaml, os.path.join(result_dir, os.path.basename(config_yaml)))
# RANDOM BO SAMPLING PREPARATIONS; 20 SAMPLES
randombo = bo.RandomBOEngine(config)
for j in range(10):
for j in range(20):
print(f"RANDOM sampling iteration {j+1}")
job_config = randombo.ask()
ptycho_engine = ptycho.FoldSlicePtychoEngine(job_config)
ptycho_engine.run()
y_value = -np.log(ptycho_engine.metric())
randombo.tell(job_config, y_value)
print('[random] TRAIN_X\n', randombo.state['train_x'])
print('[random] TRAIN_Y\n', randombo.state['train_y'])
# MAIN SINGLE OBJECTIVE BAYESIAN OPTIMIZATION
sobo = bo.SingleObjectiveBOEngine(config)
sobo.train_x = randombo.state['train_x']
sobo.train_y = randombo.state['train_y']
sobo.train_x = randombo.train_x
sobo.train_y = randombo.train_y
for j in range(config['bo']['max_iterations']):
print(f"SOBO sampling iteration {j+1}")
job_config = sobo.ask()
ptycho_engine = ptycho.FoldSlicePtychoEngine(job_config)
ptycho_engine.run(header=f"[BO {j:03d}] ")
y_value = -np.log(ptycho_engine.metric())
sobo.tell(job_config, y_value)
print('[sobo] TRAIN_X\n', sobo.train_x)
print('[sobo] TRAIN_Y\n', sobo.train_y)
if __name__ == "__main__":
if len(sys.argv) != 2: