| from pathlib import Path | |
| import sys | |
| ROOT = Path(__file__).resolve().parents[1] | |
| sys.path.insert(0, str(ROOT)) | |
| from model.atmosphericda_dmodel import QGConfig, create_dataset, load_yaml | |
| if __name__ == "__main__": | |
| config = load_yaml(ROOT / "conf/config.yaml") | |
| qg = QGConfig(**config["qg"]) | |
| settings = config["data"] | |
| output = ROOT / config["paths"]["data"] | |
| create_dataset(str(output), settings["samples"], qg, settings["seed"], | |
| settings["observations_per_batch"], settings["observation_variance"], | |
| settings["analysis_gain"], settings["format_version"]) | |
| print(f"saved {output}: state=[{settings['samples']},2,{qg.ny},{qg.nx}] observations=[{settings['samples']},12,50]") | |