moved to main.py

This commit is contained in:
2026-07-13 15:11:14 +09:00
parent 91f6b521bb
commit 2e51879931
2 changed files with 36 additions and 94 deletions
+36
View File
@@ -0,0 +1,36 @@
import os
import sys
import yaml
import bo
import ptycho
def main(config):
bo_engine = bo.RandomBOEngine(config)
max_iterations = config['bo']['max_iterations']
for _ in range(max_iterations):
job_config = bo_engine.ask()
ptycho_engine = ptycho.FoldSlicePtychoEngine(job_config)
ptycho_engine.run()
y_value = ptycho_engine.metric()
bo_engine.tell(job_config, y_value)
print(bo_engine.state['train_x'])
print(bo_engine.state['train_y'])
if __name__ == "__main__":
if len(sys.argv) != 2:
print("Usage: python bo-ptycho.py <config_yaml>")
sys.exit(1)
config_yaml = sys.argv[1]
with open(config_yaml, 'r') as f:
config = yaml.safe_load(f)
main(config)