File size: 541 Bytes
61c016d
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
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")