File size: 732 Bytes
9613b5e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | 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]")
|