From b1ac6e0be07d68b4698bcd972f1a6e0d60f6e84d Mon Sep 17 00:00:00 2001 From: OuyangWenyu Date: Mon, 10 Apr 2023 23:20:36 +0800 Subject: [PATCH] fix a sample data bug --- hydromodel/app/calibrate_xaj.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/hydromodel/app/calibrate_xaj.py b/hydromodel/app/calibrate_xaj.py index 20f198b..b0ccc88 100644 --- a/hydromodel/app/calibrate_xaj.py +++ b/hydromodel/app/calibrate_xaj.py @@ -34,10 +34,11 @@ def calibrate(args): algo_info = args.algorithm comment = args.comment data_dir = os.path.join(definitions.ROOT_DIR, "hydromodel", "example", exp) - kfold = [] - for f_name in os.listdir(data_dir): - if fnmatch.fnmatch(f_name, "*_fold*_test.json"): - kfold.append(int(f_name[len("data_info_fold") : -len("_test.json")])) + kfold = [ + int(f_name[len("data_info_fold") : -len("_test.json")]) + for f_name in os.listdir(data_dir) + if fnmatch.fnmatch(f_name, "*_fold*_test.json") + ] kfold = np.sort(kfold) for fold in kfold: print(f"Start to calibrate the {fold}-th fold")