fixes to results dir

This commit is contained in:
2026-07-21 16:04:12 +09:00
parent 99b15555db
commit 6b2f6531b8
2 changed files with 9 additions and 5 deletions
+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)