fixes to results dir

This commit is contained in:
swim committed 2026-07-21 16:04:12 +09:00
1 parent 99b15555db
commit 6b2f6531b8
2 files changed
+9 -5

No files matched your search

+1 -1
View File
@@ -1,6 +1,6 @@
io: io:
input_data_path: '/home/swim/Si_project/data/Si2V1_1.mat' input_data_path: '/home/swim/Si_project/data/Si2V1_1.mat'
result_dir: '/home/swim/Si_project/wrapper/results260721' result_dir: '/home/swim/Si_project/wrapper/results/260721'
verbosity: 1 verbosity: 1
ptycho: ptycho:
+8 -4
View File
@@ -1,15 +1,21 @@
import os import os
import sys import sys
import yaml import yaml
import shutil
import numpy as np import numpy as np
import bo import bo
import ptycho import ptycho
def main(config): def main(config_yaml):
with open(config_yaml, 'r') as f:
config = yaml.safe_load(f)
result_dir = config['io']['result_dir'] result_dir = config['io']['result_dir']
os.makedirs(result_dir, exist_ok=True) os.makedirs(result_dir, exist_ok=True)
shutil.copy(config_yaml, os.path.join(result_dir, os.path.basename(config_yaml)))
randombo = bo.RandomBOEngine(config) randombo = bo.RandomBOEngine(config)
@@ -49,7 +55,5 @@ if __name__ == "__main__":
sys.exit(1) sys.exit(1)
config_yaml = sys.argv[1] config_yaml = sys.argv[1]
with open(config_yaml, 'r') as f:
config = yaml.safe_load(f)
main(config) main(config_yaml)