GCRM-ConvectionNN / scripts /fake_data.py
zhangrenchao's picture
Upload folder using huggingface_hub
61c016d verified
Raw History Blame Contribute Delete
541 Bytes
from pathlib import Path
import sys
ROOT = Path(__file__).resolve().parents[1]
sys.path.insert(0, str(ROOT))
from model.gcrm_convectionnn import create_structured_dataset, load_yaml
if __name__ == "__main__":
config = load_yaml(ROOT / "conf/config.yaml")
output = ROOT / config["paths"]["data"]
create_structured_dataset(str(output), **config["data"], format_version=config["project"]["format_version"])
print(f"saved {output}: inputs=[{config['data']['samples']},70], target=[{config['data']['samples']},68], levels=34")