mirror of
https://github.com/c-sooyoung/bo-ptycho.git
synced 2026-09-17 23:29:07 +09:00
41 lines
1.5 KiB
Python
41 lines
1.5 KiB
Python
import os
|
|
import bo
|
|
import ptycho
|
|
|
|
def sobo_job(config):
|
|
|
|
print("Hello")
|
|
# result_dir = config['io']['result_dir']
|
|
# os.makedirs(result_dir, exist_ok=True)
|
|
|
|
# randombo = bo.RandomBOEngine(config)
|
|
|
|
# bo_txt = os.path.join(result_dir, "bo.txt")
|
|
# with open(bo_txt, "w") as f:
|
|
# f.write(f" iter\tmetric\t{"\t".join([p[:7] for p in randombo.params])}\n")
|
|
|
|
# for j in range(J):
|
|
# print(f"RANDOM sampling iteration {j}")
|
|
# job_config = randombo.ask()
|
|
# ptycho_engine = ptycho.FoldSlicePtychoEngine(job_config)
|
|
# ptycho_engine.run(run_id=f"bo-{j:03d}")
|
|
# y_value = ptycho_engine._metric
|
|
# randombo.tell(job_config, y_value)
|
|
# with open(bo_txt, "a") as f:
|
|
# p = [f'{job_config['ptycho']['params'][key]:.2f}' for key in randombo.params]
|
|
# f.write(f"{j: 8d}\t{y_value:.4f}\t{"\t".join(p)}\n")
|
|
|
|
|
|
# sobo = bo.SingleObjectiveBOEngine(config)
|
|
# sobo.train_x = randombo.train_x
|
|
# sobo.train_y = randombo.train_y
|
|
# for j in range(J, J + config['bo']['max_iterations'] + 1):
|
|
# print(f"SOBO sampling iteration {j}")
|
|
# job_config = sobo.ask()
|
|
# ptycho_engine = ptycho.FoldSlicePtychoEngine(job_config)
|
|
# ptycho_engine.run(run_id=f"bo-{j:03d}")
|
|
# y_value = ptycho_engine._metric
|
|
# sobo.tell(job_config, y_value)
|
|
# with open(bo_txt, "a") as f:
|
|
# p = [f'{job_config['ptycho']['params'][key]:.2f}' for key in sobo.params]
|
|
# f.write(f"{j: 8d}\t{y_value:.4f}\t{"\t".join(p)}\n") |